Update app.py

app.py

@@ -32,6 +32,8 @@ def classify_image(image_file):
 
     return output_image, f"Predicted class: {label} (confidence: {confidence:.2f})"
 
+from gradio import gradio as gr
+
 iface = gr.Interface(
     fn=classify_image,
     inputs=gr.inputs.Image(type="filepath", label="Upload an image"),
@@ -40,5 +42,4 @@ iface = gr.Interface(
     description="Classify images using a pre-trained ViT model",
 )
 
-# Launch the Gradio app
 iface.launch()