johann22 commited on
Commit
61000a5
·
1 Parent(s): 9fe8408

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -34,8 +34,10 @@ def gen_from_infer(purpose,history,image,model_drop,chat_drop,choice):
34
  out_prompt=generate(purpose,history,chat_drop)
35
  history.append((purpose,out_prompt))
36
  yield (history,None)
37
-
38
- infer=InferenceClient(loaded_model[int(model_drop)])
 
 
39
  out_img=infer.text_to_image(prompt=out_prompt)
40
  '''prompt=out_prompt,
41
  negative_prompt=None,
 
34
  out_prompt=generate(purpose,history,chat_drop)
35
  history.append((purpose,out_prompt))
36
  yield (history,None)
37
+ infer_model = loaded_model[int(model_drop)]
38
+ print (infer_model)
39
+ infer=InferenceClient(f'{infer_model}')
40
+ print (infer)
41
  out_img=infer.text_to_image(prompt=out_prompt)
42
  '''prompt=out_prompt,
43
  negative_prompt=None,