musdfakoc commited on
Commit
a2aee3a
·
verified ·
1 Parent(s): f36caa0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -165,7 +165,7 @@ def main():
165
  generator = Generator(output_time_frames).to(device)
166
 
167
  # Load the pre-trained model
168
- model_path = '/path/to/your/model/gan_model_100e_16b.pth' # Change this path
169
  generator = load_gan_model(generator, model_path, device)
170
 
171
  # Gradio interface: allow users to upload an image and generate audio
 
165
  generator = Generator(output_time_frames).to(device)
166
 
167
  # Load the pre-trained model
168
+ model_path = './gan_model.pth' # Change this path
169
  generator = load_gan_model(generator, model_path, device)
170
 
171
  # Gradio interface: allow users to upload an image and generate audio