Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ original_load = torch.load
|
|
| 14 |
torch.load = lambda *args, **kwargs: original_load(*args, **kwargs, map_location=torch.device('cpu'))
|
| 15 |
|
| 16 |
# Initialize the AuraSR model
|
| 17 |
-
aura_sr = AuraSR.from_pretrained("fal
|
| 18 |
|
| 19 |
# Restore original torch.load
|
| 20 |
torch.load = original_load
|
|
|
|
| 14 |
torch.load = lambda *args, **kwargs: original_load(*args, **kwargs, map_location=torch.device('cpu'))
|
| 15 |
|
| 16 |
# Initialize the AuraSR model
|
| 17 |
+
aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2")
|
| 18 |
|
| 19 |
# Restore original torch.load
|
| 20 |
torch.load = original_load
|