Update app.py

app.py

@@ -7,8 +7,8 @@ import time
 model = CustomModel()
 
 def predict(image: Image.Image):
-    # סימולציה של זמן עיבוד (למשל 5 שניות)
-    time.sleep(5)  # אפשר לשנות את הזמן לפי הצורך
+    # סימולציה של זמן עיבוד
+    time.sleep(2)  # שינוי הזמן לפי הצורך
     label, confidence = model.predict(image)
     result = "AI image" if label == 1 else "Real image"
     return result, f"Confidence: {confidence:.2f}%"
@@ -21,14 +21,16 @@ with gr.Blocks() as demo:
     output_confidence = gr.Textbox(label="Confidence", interactive=False)
 
     def show_animation(image):
-        return animation.update(visible=True), "", ""  # להציג את האנימציה
+        animation.visible = True
+        return animation, "", ""  # להציג את האנימציה
 
     def hide_animation(image):
         result, confidence = predict(image)
-        return animation.update(visible=False), result, confidence  # להסתיר את האנימציה
+        animation.visible = False  # להסתיר את האנימציה
+        return result, confidence  # להחזיר את התוצאות
 
     image_input.change(show_animation, inputs=image_input, outputs=[animation, output_label, output_confidence])
-    image_input.change(hide_animation, inputs=image_input, outputs=[animation, output_label, output_confidence])
+    image_input.submit(hide_animation, inputs=image_input, outputs=[output_label, output_confidence])
 
 # Launch the Gradio interface
 demo.launch()