Update app.py

app.py

@@ -12,17 +12,16 @@ def predict(image: Image.Image):
         animation.visible = False  # Hide the animation
         return result, f"Confidence: {confidence:.2f}%"
 
-theme = gr.themes.Soft()
 
-with gr.Interface(theme=theme) as demo:
     # 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."
-    )
+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 = gr.themes.Soft()
+)
 
 # Launch the Gradio interface
 if __name__ == "__main__":