Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 
-title = "GPT-J-6B"
+title = "Mushroom Gallery"
 
 tts_examples = [
     "I love learning machine learning",
@@ -14,6 +14,12 @@ tts_demo = gr.Interface.load(
     description="Give me something to say!",
 )
 
+vision_demo = gr.Interface.load(
+    "models/google/vit-base-patch16-224",
+    title=None,
+    description="Image Classification",
+)
+
 stt_demo = gr.Interface.load(
     "huggingface/facebook/wav2vec2-base-960h",
     title=None,
@@ -21,7 +27,7 @@ stt_demo = gr.Interface.load(
     description="Let me try to guess what you're saying!",
 )
 
-demo = gr.TabbedInterface([tts_demo, stt_demo], ["Text-to-speech", "Speech-to-text"])
+demo = gr.TabbedInterface([vision_demo, tts_demo, stt_demo], ["Image Classification", "Text-to-speech", "Speech-to-text"])
 
 if __name__ == "__main__":
     demo.launch()