Update app.py

app.py

@@ -88,11 +88,11 @@ class Generator(nn.Module):
 
 # 🔧 Load the models
 model1 = Generator(3, 1, 3)
-model1.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
+model1.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu'), weights_only=True))
 model1.eval()
 
 model2 = Generator(3, 1, 3)
-model2.load_state_dict(torch.load('model2.pth', map_location=torch.device('cpu')))
+model2.load_state_dict(torch.load('model2.pth', map_location=torch.device('cpu'), weights_only=True))
 model2.eval()
 
 # 🖼️ Function to process the image and create line drawing
@@ -124,8 +124,15 @@ for file in os.listdir(image_dir):
         examples.append([file, 'Complex Lines'])
 
 # 🚀 Create and launch the Gradio interface
-iface = gr.Interface(predict, [gr.inputs.Image(type='filepath'),
-    gr.inputs.Radio(['Complex Lines','Simple Lines'], type="value", default='Simple Lines', label='version')],
-    gr.outputs.Image(type="pil"), title=title, examples=examples)
+iface = gr.Interface(
+    fn=predict, 
+    inputs=[
+        gr.Image(type='filepath'),
+        gr.Radio(['Complex Lines', 'Simple Lines'], label='version', value='Simple Lines')
+    ],
+    outputs=gr.Image(type="pil"),
+    title=title,
+    examples=examples
+)
 
 iface.launch()