gokaygokay commited on
Commit
8bc7e88
1 Parent(s): 4479682

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-ai/AuraSR-v2")
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