Update generator_model_40.pth loading in app.py

app.py

@@ -57,7 +57,7 @@ if (device.type == 'cuda') and (ngpu > 1):
     generator = nn.DataParallel(generator, list(range(ngpu)))
     discriminator = nn.DataParallel(discriminator, list(range(ngpu)))
     seg_model = nn.DataParallel(seg_model, list(range(ngpu)))
-
+generator.load_state_dict(torch.load('generator_model_40.pth'))
 
 def inference(sketch_path,label):
     transform_sketch = T.Compose(
@@ -79,8 +79,8 @@ def inference(sketch_path,label):
     return fake_images.cpu().detach().numpy()
 
 
-audio_1 = gr.Image(sources="upload", type="filepath", label="Audio 1")
-audio_2 = gr.Image(sources="upload", type="filepath", label="Audio 2")
+audio_1 = gr.Image(sources="upload", type="filepath", label="img 1")
+audio_2 = gr.Image(sources="upload", type="filepath", label="img 2")
 # text_output = gr.Textbox(label="Similarity Score")
 image_out = gr.Image(label="Generated Image")
 gr.Interface(