uhdessai commited on
Commit
2c5d7c0
·
verified ·
1 Parent(s): 9a318ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -133,7 +133,7 @@ import tempfile
133
  device = torch.device("cpu")
134
 
135
  # Load your GAN model
136
- with open("dress_model.pkl", "rb") as f:
137
  G = pickle.load(f)['G_ema'].eval().cpu() # Ensure model is in eval mode and on CPU
138
 
139
  # Load CLIP model and processor
 
133
  device = torch.device("cpu")
134
 
135
  # Load your GAN model
136
+ with open("dress.pkl", "rb") as f:
137
  G = pickle.load(f)['G_ema'].eval().cpu() # Ensure model is in eval mode and on CPU
138
 
139
  # Load CLIP model and processor