ArmelR commited on
Commit
bcf574f
1 Parent(s): c2b8350

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -104,21 +104,20 @@ def generate(
104
  else :
105
  ValueError("Unsupported version of the Coding assistant")
106
 
107
- print([response.token.text for response in stream])
108
  output = ""
109
  previous_token = ""
110
 
111
  for response in stream:
112
  if (
113
- (response.token.text in ["Human", "-----", "Question:"]
114
- and previous_token in ["\n", "-----"])
115
  or response.token.text in ["<|endoftext|>", "<|end|>", "Answer:"]
116
  ):
117
  return output.strip()
118
  else:
119
  output += response.token.text
120
  previous_token = response.token.text
121
-
122
  return output.strip()
123
 
124
 
 
104
  else :
105
  ValueError("Unsupported version of the Coding assistant")
106
 
107
+ print("Tokens = "+str([response.token.text for response in stream]))
108
  output = ""
109
  previous_token = ""
110
 
111
  for response in stream:
112
  if (
113
+ (response.token.text in ["Human", "-----", "Question:"] and previous_token in ["\n", "-----"])
 
114
  or response.token.text in ["<|endoftext|>", "<|end|>", "Answer:"]
115
  ):
116
  return output.strip()
117
  else:
118
  output += response.token.text
119
  previous_token = response.token.text
120
+ print("Output = "+str(output.strip()))
121
  return output.strip()
122
 
123