Minh Nguyen commited on
Commit
fde940c
·
1 Parent(s): 747e34b
Files changed (2) hide show
  1. app.py +3 -4
  2. requirements.txt +2 -1
app.py CHANGED
@@ -60,10 +60,9 @@ def respond(
60
  response = ""
61
 
62
  for message in text_streamer:
63
- # not append if message contain <|eot_id|>
64
- if "<|eot_id|>" not in message:
65
- response += message
66
- yield response
67
 
68
 
69
  """
 
60
  response = ""
61
 
62
  for message in text_streamer:
63
+ # remove <|eot_id|>
64
+ response += message.split("<|eot_id|>")[0]
65
+ yield response
 
66
 
67
 
68
  """
requirements.txt CHANGED
@@ -1 +1,2 @@
1
- huggingface_hub==0.25.2
 
 
1
+ huggingface_hub==0.25.2
2
+ unsloth