Update app.py

app.py

@@ -31,6 +31,7 @@ history = []
 
 def gen_from_infer(purpose,history,image,model_drop,chat_drop,choice):
     #out_img = infer(out_prompt)
+    history = ""
     out_prompt=generate(purpose,history,chat_drop)
     history.append((purpose,out_prompt))
     yield (history,None)        
@@ -38,33 +39,19 @@ def gen_from_infer(purpose,history,image,model_drop,chat_drop,choice):
     print (infer_model)
     infer=InferenceClient(f'{infer_model}')
     print (infer)
-    out_img=infer.text_to_image(prompt=out_prompt)
-    '''prompt=out_prompt,
-    negative_prompt=None,
-    height=512,
-    width=512,
-    num_inference_steps=None,
-    guidance_scale=None,
-    model=,
-    seed=1,
-    )'''
+    
+    out_img=infer.text_to_image(
+        prompt=out_prompt,
+        negative_prompt=None,
+        height=512,
+        width=512,
+        num_inference_steps=None,
+        guidance_scale=None,
+        model=None,
+        seed=1,
+        )
     yield (history,out_img)
-    '''
-    #return (history,None)
-    print(out_img)
-    url=f'https://johann22-chat-diffusion-describe.hf.space/file={out_img}'
-    print(url)
-    uid = uuid.uuid4()
-    #urllib.request.urlretrieve(image, 'tmp.png')
-    #out=Image.open('tmp.png')
-    r = requests.get(url, stream=True)
-    if r.status_code == 200:
-        out = Image.open(io.BytesIO(r.content))
-        #yield ([(purpose,out_prompt)],out)
-        yield (history,out)
-    else:
-        yield ([(purpose,"an Error occured")],None)
-    '''
+
 def format_prompt(message, history):
   prompt = "<s>"
   for user_prompt, bot_response in history: