remove diffusion

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 import os
 import openai
-from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
+#from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 import torch 
 
 model_id = "stabilityai/stable-diffusion-2-1"
-pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
+#pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
+
 # pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 # pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 # pipe = pipe.to("cuda")
@@ -33,9 +34,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,
@@ -46,7 +47,7 @@ demo = gr.Interface(
         # gr.Dropdown(team_list, value=[team_list[random.randint(1,30)]], multiselect=True),
         # gr.Checkbox(label="Is it the morning?"),
     ],
-    ["text","image"],
+    ["text"],#,"image"],
     # "image",
     allow_flagging="never",
     title="GPT-3 Analogy Lab 🧪",