Spaces:

IEIT-Yuan
/

Yuan2-2B-demo

Runtime error

IEIT-Yuan commited on Jan 15, 2024

Commit

5dc5c4e

verified ·

1 Parent(s): 28f4c11

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,6 +60,7 @@ def predict(message, history):
     #     yield partial_message
     inputs = tokenizer(message, return_tensors="pt")["input_ids"].to(device)
     outputs = model.generate(inputs, do_sample=False, max_length=100)
     return(tokenizer.decode(outputs[0]))

     #     yield partial_message
     inputs = tokenizer(message, return_tensors="pt")["input_ids"].to(device)
     outputs = model.generate(inputs, do_sample=False, max_length=100)
+    print(tokenizer.decode(outputs[0]))
     return(tokenizer.decode(outputs[0]))