Update app.py

app.py

@@ -35,12 +35,8 @@ def calculate_similarity(image1, image2):
         # Analyze the two cropped images for facial similarity
         result = DeepFace.verify(face1_rgb, face2_rgb, model_name='VGG-Face', enforce_detection=False)
         similarity_percentage = (1 - result['distance']) * 100  # Convert distance to similarity percentage
-
-        # Encode images to display in Gradio
-        _, face1_encoded = cv2.imencode('.png', face1)
-        _, face2_encoded = cv2.imencode('.png', face2)
-
-        return similarity_percentage, face1_encoded.tobytes(), face2_encoded.tobytes()
+        
+        return similarity_percentage, face1_rgb, face2_rgb
     except Exception as e:
         return str(e), None, None
 
@@ -53,8 +49,8 @@ iface = gr.Interface(
     ],
     outputs=[
         gr.outputs.Textbox(label="Similarity Percentage"),
-        gr.outputs.Image(label="Cropped Face 1"),
-        gr.outputs.Image(label="Cropped Face 2"),
+        gr.outputs.Image(label="Cropped Face 1", type="numpy"),
+        gr.outputs.Image(label="Cropped Face 2", type="numpy"),
     ],
     title="Face Similarity Checker",
     description="Upload two images of faces to check their similarity. The faces will be detected and cropped automatically."