Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -131,7 +131,7 @@ import gradio as gr
|
|
| 131 |
device = torch.device("cpu")
|
| 132 |
|
| 133 |
# Load your GAN model
|
| 134 |
-
with open("
|
| 135 |
G = pickle.load(f)['G_ema'].eval().cpu() # Ensure model is in eval mode and on CPU
|
| 136 |
|
| 137 |
# Load CLIP model and processor
|
|
|
|
| 131 |
device = torch.device("cpu")
|
| 132 |
|
| 133 |
# Load your GAN model
|
| 134 |
+
with open("blazer_model.pkl", "rb") as f:
|
| 135 |
G = pickle.load(f)['G_ema'].eval().cpu() # Ensure model is in eval mode and on CPU
|
| 136 |
|
| 137 |
# Load CLIP model and processor
|