Update app.py

app.py

@@ -5,17 +5,16 @@ from vit_model_test import CustomModel
 # Initialize the model
 model = CustomModel()
 
-theme = gr.themes.Base()
-
-with gr.Blocks(theme=theme) as demo:
-
-    def predict(image: Image.Image):
+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}%"
 
+theme = gr.themes.Base()
+
+with gr.Blocks(theme=theme) as demo:
     # Define the Gradio interface
     gr.Interface(
         fn=predict,