Update app.py

app.py

@@ -41,12 +41,12 @@ class Predictor:
             print("No face found")
             return None
 
-    def predict(self, input_image, swap_image):
+    def predict(self, input_image_path, swap_image_path):
         """Run a single prediction on the model"""
         try:
-            frame = cv2.imread(input_image.name)
+            frame = cv2.imread(input_image_path)
             face = self.get_face(frame)
-            source_face = self.get_face(cv2.imread(swap_image.name))
+            source_face = self.get_face(cv2.imread(swap_image_path))
             try:
                 print(frame.shape, face.shape, source_face.shape)
             except:
@@ -78,8 +78,8 @@ iface = gr.Interface(
     ],
     outputs=gr.Image(type="filepath", label="Result"),
     title=title,
-    examples=[["input.jpg", "swap img.jpg"]])
-
+    examples=[["input.jpg", "swap img.jpg"]]
+)
 
 # Launch the Gradio Interface
 iface.launch()