Update app.py

app.py

@@ -1,27 +1,55 @@
 import gradio as gr
 from PIL import Image
 from vit_model_test import CustomModel
+import time
 
 # Initialize the model
 model = CustomModel()
 
 def predict(image: Image.Image):
-    
+    time.sleep(5)  # סימולציה של זמן עיבוד
     label, confidence = model.predict(image)
     result = "AI image" if label == 1 else "Real image"
     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.",
-    theme="compact",
-    live=True
-)
-
-# Launch the Gradio interface
-demo.launch()
+
+def loading_animation(image):
+    return gr.Video.update(visible=True), "", ""
+
+def show_results(image):
+    result, confidence = predict(image)
+    return gr.Video.update(visible=False), result, confidence
+
+# יצירת ממשק Gradio עם אנימציות
+with gr.Blocks(css="""
+    .gr-button { 
+        background-color: #4CAF50; 
+        color: white; 
+        transition: background-color 0.3s ease;
+    }
+    .gr-button:hover { 
+        background-color: #45a049; 
+    }
+    .fade-in { 
+        animation: fadeIn 0.5s ease-in; 
+    }
+    @keyframes fadeIn { 
+        from { opacity: 0; } 
+        to { opacity: 1; } 
+    }
+""") as demo:
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="Upload an image")
+        animation = gr.Video("https://cdn-uploads.huggingface.co/production/uploads/66d6f1b3b50e35e1709bfdf7/x7Ud8PO9QPfmrTvBVcCKE.mp4", visible=False)
+        
+        output_label = gr.Textbox(label="Classification Result", interactive=False, visible=False)
+        output_confidence = gr.Textbox(label="Confidence", interactive=False, visible=False)
+
+    image_input.change(loading_animation, inputs=image_input, outputs=[animation, output_label, output_confidence])
+    image_input.change(show_results, inputs=image_input, outputs=[animation, output_label, output_confidence])
+
+    # הוספת כפתור עם אנימציה
+    submit_button = gr.Button("Submit", css="fade-in")  # כפתור עם אפקט של fade-in
+    submit_button.click(predict, inputs=image_input, outputs=[output_label, output_confidence])
+
+# השקת הממשק
+demo.launch()