Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from transformers import pipeline
 
-vision_classifier = pipeline(task="image-classification",
-                             model="microsoft/beit-base-patch16-224-pt22k-ft22k")
-iface = gr.Interface.from_pipeline(vision_classifier, 
-                                   title="2-line Object Recognition using HF Pipeline",
-                                   description="Demonstrates how to use HF pipeline and gradio.",
-                                   examples=['data/wonder_cat.jpg', 'data/aki_dog.jpg',],
-                                   article = "Author: <a href=\"https://huggingface.co/rowel\">Rowel Atienza</a>",
-                                   ).launch()
+pipe = pipeline(task="image-classification", 
+                # model that can do 22k-category classification
+                model="microsoft/beit-base-patch16-224-pt22k-ft22k")
+gr.Interface.from_pipeline(pipe, 
+                           title="22k Image Classification",
+                           description="Object Recognition using Microsoft BEIT (22k classes)",
+                           examples = ['wonder_cat.jpg', 'aki_dog.jpg',],
+                           ).launch()