Update app.py

app.py

@@ -75,10 +75,34 @@ def ner_drugs(text):
 med_iface = gr.Interface(fn=ner_drugs,inputs = "text",outputs=["text"],title="Drugs Named Entity Recognition").queue()
 
 
+from diffusers import StableDiffusionPipeline
+
+#pip install accelerate
+
+#pip install --user 
+
+#pip install --user torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
+
+pipe = StableDiffusionPipeline.from_pretrained('CompVis/stable-diffusion-v1-4')
+
+
+
+# Initialize a prompt
+def stable_image(text):
+  prompt = text
+  # Pass the prompt in the pipeline
+  return pipe(prompt).images[0]
+
+
+
+import gradio as gr
+stable_iface = gr.Interface(fn=stable_image, inputs= "text",outputs=["image"],title="Text to Image").queue()
+
+
 
 demo = gr.TabbedInterface(
 
-    [txtgen_iface, sum_iface, med_iface], ["Text Generation", "Summary Generation", "Drug Named-entity recognition"],
+    [txtgen_iface, sum_iface, med_iface,stable_iface], ["Text Generation", "Summary Generation", "Drug Named-entity recognition","Text to Image"],
     title="All the Models of UWC",
 
 )