Update app.py

app.py

@@ -9,26 +9,22 @@ theme = gr.themes.Base()
 
 with gr.Blocks(theme=theme) as demo:
 
-def predict(image: Image.Image):
-    animation.visible = True  # הראה את האנימציה
-    label, confidence = model.predict(image)
-    result = "AI image" if label == 1 else "Real image"
-    animation.visible = False  # הסתר את האנימציה
-    return result, f"Confidence: {confidence:.2f}%"
-
-
-# Define the Gradio interface
-demo = gr.Interface(
-    fn=predict,
-    inputs=gr.Image(type="pil"),
-    outputs=[gr.Textbox(), gr.Textbox()],
-    title="Vision Transformer Model",
-    description="Upload an image to classify it using the Vision Transformer model.",
-    live=True,  # מאפשר חיזוי מיידי
-)
-
-# Add video component
-animation = gr.Video("https://cdn-uploads.huggingface.co/production/uploads/66d6f1b3b50e35e1709bfdf7/x7Ud8PO9QPfmrTvBVcCKE.mp4", visible=True)  # הכניסי את ה-URL הנכון
+    def predict(image: Image.Image):
+        animation.visible = True  # Show the animation
+        label, confidence = model.predict(image)
+        result = "AI image" if label == 1 else "Real image"
+        animation.visible = False  # Hide the animation
+        return result, f"Confidence: {confidence:.2f}%"
+
+    # Define the Gradio interface
+    gr.Interface(
+        fn=predict,
+        inputs=gr.Image(type="pil"),
+        outputs=[gr.Textbox(), gr.Textbox()],
+        title="Vision Transformer Model",
+        description="Upload an image to classify it using the Vision Transformer model.",
+        live=True,  # Allows immediate prediction
+    )
 
 # Launch the Gradio interface
 if __name__ == "__main__":