Update app.py

app.py

@@ -120,14 +120,14 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
     #jobj.close()    
     #chat_im_out=chat_img(output)
 
-def gen_im(prompt,seed):
+'''def gen_im(prompt,seed):
     print('generating image')
     image_out = im_client.text_to_image(prompt=prompt['text'],height=128,width=128,num_inference_steps=10,seed=seed)
     #print(type(image_out))
     output=f'images/{uuid.uuid4()}.png'
     image_out.save(output)
     print('Done: ',output)
-    return [{'role':'assistant','content': {'path':output}}]
+    return [{'role':'assistant','content': {'path':output}}]'''
 
 def build_space(repo_name,file_name,file_content,access_token=""):
     try:
@@ -200,8 +200,8 @@ def agent(prompt_in,history,mod=2):
                     build_space(out_w[0],out_w[1],out_w[2])
                 elif 'IMAGE' in fn:
                     print('IMAGE called')
-                    out_im=gen_im(prompt,seed)
-                    yield [{'role':'assistant','content': out_im}]
+                    #out_im=gen_im(prompt,seed)
+                    #yield [{'role':'assistant','content': out_im}]
                 elif 'SEARCH' in fn:
                     print('SEARCH called')
                 elif 'COMPLETE' in fn: