Update app.py

app.py

@@ -1,7 +1,8 @@
+# Install required packages
 !pip install transformers
 !pip install streamlit
 !pip install torch
-!pip install transformers torch datasets gradio
+!pip install datasets gradio
 
 import gradio as gr
 from transformers import pipeline
@@ -18,11 +19,11 @@ def classify_image(image):
 interface = gr.Interface(
     fn=classify_image,
     inputs=gr.Image(type="pil"),
-    outputs=gr.Label(),
+    outputs=gr.JSON(),  # Updated to JSON for dictionary-like output
     title="Image Classification App",
     description="Upload an image, and the app will classify it using a Vision Transformer (ViT) model."
 )
 
 # Launch the app
-if _name_ == "_main_":
-    interface.launch()
+if __name__ == "__main__":
+    interface.launch()