Spaces:

ThongCoding
/

Gradio

Runtime error

ThongCoding commited on 3 days ago

Commit

ed7d958

verified ·

1 Parent(s): beb797f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ def respond(message, history):
     # encode + generate
     inputs  = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
-        **inputs,
         max_new_tokens=200,
         do_sample=False,     # greedy
         pad_token_id=tokenizer.eos_token_id

     # encode + generate
     inputs  = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
+        inputs[0],
         max_new_tokens=200,
         do_sample=False,     # greedy
         pad_token_id=tokenizer.eos_token_id