Update app.py

app.py

@@ -1,7 +1,25 @@
-from transformers import AutoTokenizer, AutoModel
+import gradio as gr
+from transformers import pipeline
 
-tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
-model = AutoModel.from_pretrained("bert-base-uncased")
+# Load the Stable Diffusion TTS model
+tts_demo = gr.Interface.load(
+    "models/runwayml/stable-diffusion-v1-5",
+    title=None,
+    description="Give me something to paint!"
+)
 
-inputs = tokenizer("Hello world!", return_tensors="pt")
-outputs = model(**inputs)
+# Load the text-to-image prompt generator model
+stt_demo = gr.Interface.load(
+    "models/succinctly/text2image-prompt-generator",
+    title=None,
+    description="Let me try to guess what you're saying!"
+)
+
+# Load the image classification model
+image_classification = pipeline("image-classification")
+
+# Create the tabbed interface
+demo = gr.TabbedInterface([tts_demo, stt_demo, gr.Interface.from_pipeline(image_classification)], ["Paint", "Text", "Image"])
+
+# Launch the interface
+demo.launch()