ldhldh commited on
Commit
5ec78b2
ยท
1 Parent(s): bfae98b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -69,8 +69,7 @@ def chat(id, npc, prompt):
69
 
70
  # model inference
71
 
72
- init()
73
- if check:
74
  prom = ""
75
  inputs = tokenizer(prom, return_tensors="pt")["input_ids"]
76
  outputs = model.generate(inputs, max_new_tokens=100)
 
69
 
70
  # model inference
71
 
72
+ if check(model_name):
 
73
  prom = ""
74
  inputs = tokenizer(prom, return_tensors="pt")["input_ids"]
75
  outputs = model.generate(inputs, max_new_tokens=100)