Update app.py

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
+from transformers import pipeline
 
-gr.Interface.load(
-             "huggingface/google/vit-base-patch16-224-in21k",
+pipe = pipeline(task="image-classification", 
+                # model that can do 22k-category classification
+                model="huggingface/google/vit-base-patch16-224-in21k")
+
+gr.Interface.load(pipe,
              theme="default",
              examples=[["alligator.jpg"], ["laptop.jpg"]],
              css=".footer{display:none !important}",