Update app.py

app.py

@@ -1,17 +1,12 @@
 import gradio as gr
 from transformers import pipeline
 
-# Define the full path to the NeMo Canary model
-model_name = "nvidia/canary-1b"
-
-# Define the pipeline for speech recognition with the NeMo Canary model
-pipe = pipeline(
-    task="automatic-speech-recognition", 
-    model=model_name
-)
-
-# Create a Gradio interface from the pipeline
-interface = gr.Interface.from_pipeline(pipe, inputs="microphone", outputs="text")
-
-# Launch the interface
-interface.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",
+                           examples = ['wonder_cat.jpg', 'aki_dog.jpg',],
+                           article = "Author: <a href=\"https://huggingface.co/rowel\">Rowel Atienza</a>",
+                           ).launch()