Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -133,7 +133,7 @@ import tempfile
|
|
| 133 |
device = torch.device("cpu")
|
| 134 |
|
| 135 |
# Load your GAN model
|
| 136 |
-
with open("
|
| 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
|