Update app.py

app.py

@@ -1,30 +1,26 @@
+import gradio as gr
 from transformers import ViTForImageClassification, ViTImageProcessor
-import torch
 from PIL import Image
-import gradio as gr
-import os
 
-# Load model
+# Load a simple AI model
 model_name = "google/vit-base-patch16-224"
 processor = ViTImageProcessor.from_pretrained(model_name)
 model = ViTForImageClassification.from_pretrained(model_name)
 
 def detect_deepfake(image):
-    if image.mode != 'RGB':
-        image = image.convert('RGB')
+    # Check if image is fake
     inputs = processor(images=image, return_tensors="pt")
-    with torch.no_grad():
-        outputs = model(**inputs)
-    return "Real" if outputs.logits.argmax() == 0 else "Fake"
+    outputs = model(**inputs)
+    return "REAL" if outputs.logits.argmax() == 0 else "FAKE (Deepfake!)"
 
-# Create interface WITHOUT examples
-iface = gr.Interface(
+# Create a simple web interface
+app = gr.Interface(
     fn=detect_deepfake,
-    inputs=gr.Image(type="pil"),
-    outputs="text",
-    title="Deepfake Detection",
-    description="Upload an image to check if it's real or AI-generated"
+    inputs=gr.Image(type="pil", label="Upload Image"),
+    outputs=gr.Label(label="Result"),
+    title="🔍 Deepfake Detector",
+    description="Upload a face photo to check if it's real or AI-generated!"
 )
 
-if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+# Launch the app with a public URL
+app.launch(share=True)  # << This gives you a temporary public link!