Update app.py

app.py

@@ -34,17 +34,18 @@ def gen_from_infer(purpose,history,image,model_drop,chat_drop,choice):
     out_prompt=generate(purpose,history,chat_drop)
     history.append((purpose,out_prompt))
     yield (history,None)        
-
-    out_img=InferenceClient().text_to_image(
-    prompt=out_prompt,
+    
+    infer=InferenceClient(loaded_model[int(model_drop)])
+    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=loaded_model[int(model_drop)],
+    model=,
     seed=1,
-    )
+    )'''
 
     #return (history,None)
     print(out_img)