Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = '
|
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
|