try diffusion

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
 import os
 import openai
-# from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
-# import torch 
+from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
+import torch 
 
-# model_id = "stabilityai/stable-diffusion-2-1"
-# pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-# pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+model_id = "stabilityai/stable-diffusion-2-1"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 # pipe = pipe.to("cuda")
 
 openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -32,9 +32,9 @@ def generate_prompt(radio,word1,word2):
     )['choices'][0]['text']
 
     response_txt = response.replace('\n','')
-    # diffusion_in = f'a dramatic painting of: {response_txt.split(".")[0]}'
-    # image = pipe(diffusion_in).images[0]
-    return response_txt#, image
+    diffusion_in = f'a dramatic painting of: {response_txt.split(".")[0]}'
+    image = pipe(diffusion_in).images[0]
+    return response_txt, image
 
 demo = gr.Interface(
     generate_prompt,
@@ -45,11 +45,11 @@ demo = gr.Interface(
         # gr.Dropdown(team_list, value=[team_list[random.randint(1,30)]], multiselect=True),
         # gr.Checkbox(label="Is it the morning?"),
     ],
-    "text",
+    ["text","image"],
     # "image",
     allow_flagging="never",
     title="GPT-3 Analogy Lab 🧪",
-    description="foo tested",
+    description="Enter two things you want to connect.",
     css="footer {visibility: hidden}"
 )