Updated app.py line 185

app.py

@@ -183,17 +183,8 @@ openai.api_key = os.environ['OPENAI_KEY']
 PROMPT = "colorful portrait 25 year bespectacled woman with long, curly skyblue hair and bright green eyes. She has a small, upturned nose and a freckled complexion. She is approximately 5'5 tall and has a thin build"
 def generate(PROMPT,model):
 #     PROMPT = "An eco-friendly computer from the 90s in the style of vaporwave""Dall-E","StableDiffusion"
-  if model=="Dall-E":
-    response = openai.Image.create(
-    prompt=PROMPT,
-    n=1,
-    size="256x256",
-)
-    x = response["data"][0]["url"]
-    urllib.request.urlretrieve(x,"file")
-    img = Image.open("file")
-  else:
-    img = generate_stability(PROMPT)
+  
+  img = generate_stability(PROMPT)
   return np.asarray(img)
 
 
@@ -201,12 +192,7 @@ def generate(PROMPT,model):
 API_ENDPOINT = "https://api.imgbb.com/1/upload"
 API_KEY = "51e7720f96af8eb5179e772e443c0c1e"
 
-# path_raw = input("Enter the image path: ")
-# path = Path(path_raw)
-# path = path_raw.replace("\\", "/")
 
-# def imgLink(path):
-# path = Path("/content/drive/MyDrive/Salz/dannydevito.png")
 def imgLink(image):
     pil_image = image.convert('RGB') 
     open_cv_image = np.array(pil_image) 
@@ -524,10 +510,8 @@ home""")
   abtn.click(generate_prompt, inputs=[text1_1,text1_2,text1_3,text1_4], outputs=text2)
   with gr.Accordion("Generate MugShot",open=False):
     gr.Markdown("**Generate MugShot from the input prompt using Dall-E**")
-    gr.Markdown("**Use Dall E  or StableDiffusion Image Generation for text to image**")
-    model = gr.Radio(["StableDiffusion"])
-    print(dir(model))
-    #with open('/content/logging.log',mode='a') as log_f: log_f.write(f'{datetime.now()} Using model for General Mugshot: {model.value}'+'\n')
+    gr.Markdown("**Use StableDiffusion Image Generation for text to image**")
+    # model = gr.Radio(["StableDiffusion"])
     with gr.Row():  
       with gr.Column():
         # seed = gr.Text(label="Input Phrase")
@@ -536,7 +520,7 @@ home""")
         # seed = gr.Text(label="Input Phrase")
         im1 = gr.Image()
     bbtn = gr.Button("Image from description")
-    bbtn.click(generate, inputs=[text3,model], outputs=im1)
+    bbtn.click(generate, inputs=[text3], outputs=im1)
 
   with gr.Accordion("Image from Sketch",open=False):
     gr.Markdown("**Get Enhanced Image from sketch and desired input promt using StableDiffusion**")