Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -69,8 +69,7 @@ def chat(id, npc, prompt):
|
|
69 |
|
70 |
# model inference
|
71 |
|
72 |
-
|
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)
|