uhdessai commited on
Commit
13824b6
·
verified ·
1 Parent(s): 93a6649

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("/content/drive/MyDrive/clothGAN/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
 
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