Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|