Update app.py

app.py

@@ -6,10 +6,8 @@ from vit_model_test import CustomModel
 model = CustomModel()
 
 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}%"