musdfakoc commited on
Commit
c812d04
·
verified ·
1 Parent(s): 6a1e7fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -179,7 +179,7 @@ def main():
179
  generator = Generator(output_time_frames).to(device)
180
 
181
  # Load the pre-trained model
182
- model_path = '/gan_model.pth' # Change this path
183
  generator = load_gan_model(generator, model_path, device)
184
 
185
  # Gradio interface: allow users to upload an image and generate audio
 
179
  generator = Generator(output_time_frames).to(device)
180
 
181
  # Load the pre-trained model
182
+ model_path = './gan_model.pth' # Change this path
183
  generator = load_gan_model(generator, model_path, device)
184
 
185
  # Gradio interface: allow users to upload an image and generate audio