johann22 commited on
Commit
97c0552
·
1 Parent(s): 61000a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ 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
- infer_model = loaded_model[int(model_drop)]
38
  print (infer_model)
39
  infer=InferenceClient(f'{infer_model}')
40
  print (infer)
 
34
  out_prompt=generate(purpose,history,chat_drop)
35
  history.append((purpose,out_prompt))
36
  yield (history,None)
37
+ infer_model = models[int(model_drop)]
38
  print (infer_model)
39
  infer=InferenceClient(f'{infer_model}')
40
  print (infer)