demo

app.py

@@ -1,8 +1,20 @@
+import os
+
 import gradio as gr
+from transformers import pipeline
+
+
+sd_description = "文字生成图片"
+sd_examples = [["小猫"], ["cat"], ["dog"]]
+sd_demo = gr.Interface.load("models/runwayml/stable-diffusion-v1-5", title='文字生成图片', examples=sd_examples)
+
+
+pipe = pipeline("image-classification")
+examples = [[os.path.join(os.path.dirname(__file__), "lion.jpg")], [os.path.join(os.path.dirname(__file__), "cat.jpeg")]]
+app = gr.Interface.from_pipeline(pipe, examples=examples, title='图片识别')
+
 
+demo = gr.TabbedInterface([sd_demo, app], ["文字生成图片", "图片识别"])
 
-description = "文字生成图片"
-examples = [["小猫"], ["cat"], ["dog"]]
 
-demo = gr.Interface.load("models/runwayml/stable-diffusion-v1-5", description=description, examples=examples)
 demo.launch()