Jezia commited on
Commit
8303ca6
·
1 Parent(s): 5d7d528

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def load_model(file_name: str, device: torch.device):
44
 
45
  local_path = '/content/'f'{file_name}'
46
  print(local_path)
47
- with open(fetch_model(local_path), 'rb') as f:
48
  model = pickle.load(f)['G_ema']
49
  model.eval()
50
  model.to(device)
 
44
 
45
  local_path = '/content/'f'{file_name}'
46
  print(local_path)
47
+ with open(local_path, 'rb') as f:
48
  model = pickle.load(f)['G_ema']
49
  model.eval()
50
  model.to(device)