uhdessai commited on
Commit
fa95816
·
verified ·
1 Parent(s): 61e4317

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -194,7 +194,7 @@ warnings.filterwarnings("ignore")
194
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
195
 
196
  # Load pretrained StyleGAN model
197
- model_path = 'dress_model.pkl'
198
  with open(model_path, 'rb') as f:
199
  G = legacy.load_network_pkl(f)['G_ema'].to(device)
200
 
 
194
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
195
 
196
  # Load pretrained StyleGAN model
197
+ model_path = 'https://drive.google.com/file/d/12_fsSQgUfOCAPQaDtq2QPCLE74qTQEwt/view?usp=sharing'
198
  with open(model_path, 'rb') as f:
199
  G = legacy.load_network_pkl(f)['G_ema'].to(device)
200