Spaces:

nroggendorff
/

dolphin

Paused

nroggendorff commited on Aug 11, 2024

Commit

699a605

verified ·

1 Parent(s): 9a79b8f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,6 +32,6 @@ def predict(input_text, history):
     outputs = model.generate(**inputs, max_new_tokens=512)
     generated_text = tokenizer.batch_decode(outputs)[0]
-    return generated_text#.split("<|assistant|>")[-1]
 gr.ChatInterface(predict, theme="soft").launch()

     outputs = model.generate(**inputs, max_new_tokens=512)
     generated_text = tokenizer.batch_decode(outputs)[0]
+    return generated_text.split("<|assistant|>")[-1]
 gr.ChatInterface(predict, theme="soft").launch()