Adding expecteted labels

app.py

@@ -49,7 +49,7 @@ data_transform = transforms.Compose([
 ])
 
 # Prediction function
-def predict(image, mode):
+def predict(image, mode, expected_output=None):
     device = 'cpu'
 
     # Apply transformations to the input image
@@ -72,13 +72,18 @@ def predict(image, mode):
         elapsed_time = time.time() - start_time
 
     predicted_class = class_names[preds[0]]
-    return f"Predicted: {predicted_class}", f"Time taken: {elapsed_time:.2f} seconds"
+    
+    # Compare predicted and expected output
+    expected_output_message = f"Expected: {expected_output}" if expected_output else "Expected: Not Provided"
+    predicted_output_message = f"Predicted: {predicted_class}"
+    
+    return predicted_output_message, expected_output_message, f"Time taken: {elapsed_time:.2f} seconds"
 
 
-# Path to example images for "Fake" and "Real" classes
+# Path to example images for "Fake" and "Real" classes along with expected outputs
 example_images = [
-    ["./data/fake/fake_1.jpeg", "Fast"],   # Fake example
-    ["./data/real/real_1.jpg", "Fast"],   # Real example
+    ["./data/fake/fake_1.jpeg", "Fake", "Fast"],   # Fake example
+    ["./data/real/real_1.jpg", "Real", "Fast"],   # Real example
 ]
 
 # Gradio interface
@@ -86,15 +91,20 @@ iface = gr.Interface(
     fn=predict,
     inputs=[
         gr.Image(type="filepath", label="Upload an Image"),  # Image input
-        gr.Radio(choices=["Fast", "Secure"], label="Inference Mode", value="Fast")  # Inference mode
+        gr.Radio(choices=["Fast", "Secure"], label="Inference Mode", value="Fast"),  # Inference mode
+        gr.Textbox(label="Expected Output", value=None, placeholder="Optional: Enter expected output (Fake/Real)")  # Expected output (optional)
     ],
     outputs=[
         gr.Textbox(label="Prediction"),  # Prediction output
+        gr.Textbox(label="Expected Output"),  # Expected output for comparison
         gr.Textbox(label="Time Taken")  # Time taken output
     ],
-    examples=example_images, 
+    examples=[  # Include expected outputs in examples
+        ["./data/fake/fake_1.jpeg", "Fast", "Fake"],  # Fake example with expected output
+        ["./data/real/real_1.jpg", "Fast", "Real"],   # Real example with expected output
+    ],
     title="Deepfake Detection Model",
-    description="Upload an image or select a sample and choose the inference mode (Fast or Secure)."
+    description="Upload an image or select a sample and choose the inference mode (Fast or Secure). Compare the predicted result with the expected output."
 )
 
 if __name__ == "__main__":