Spaces:

givyboy
/

mental-health-chatbot

Sleeping

Anthony G commited on Jan 21, 2024

Commit

60a302d

1 Parent(s): 9ff82a3

slight change

Files changed (1) hide show

app.py CHANGED Viewed

@@ -75,7 +75,6 @@ def get_model_response(message: str, history: list[str]) -> str:
         num_return_sequences=1,
         eos_token_id=peft_tokenizer.eos_token_id,
         max_length=600,
-        truncation=True,
     )[0]
     print(sequences["generated_text"])
     output = sequences["generated_text"].split("<ASSISTANT>:")[-1].strip()

         num_return_sequences=1,
         eos_token_id=peft_tokenizer.eos_token_id,
         max_length=600,
     )[0]
     print(sequences["generated_text"])
     output = sequences["generated_text"].split("<ASSISTANT>:")[-1].strip()