Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -105,11 +105,12 @@ def generate(
|
|
105 |
else :
|
106 |
ValueError("Unsupported version of the Coding assistant")
|
107 |
|
108 |
-
print("Tokens = "+str([response.token.text for response in stream]))
|
109 |
output = ""
|
110 |
previous_token = ""
|
111 |
t = 0
|
112 |
for response in stream:
|
|
|
113 |
if (
|
114 |
(response.token.text in ["Human", "-----", "Question:"] and previous_token in ["\n", "-----"])
|
115 |
or response.token.text in ["<|endoftext|>", "<|end|>", "Answer:"]
|
@@ -121,7 +122,7 @@ def generate(
|
|
121 |
print(f"Out_{t} : {output}")
|
122 |
t += 1
|
123 |
previous_token = response.token.text
|
124 |
-
print("Output = "+str(output
|
125 |
return output.strip()
|
126 |
|
127 |
|
|
|
105 |
else :
|
106 |
ValueError("Unsupported version of the Coding assistant")
|
107 |
|
108 |
+
#print("Tokens = "+str([response.token.text for response in stream]))
|
109 |
output = ""
|
110 |
previous_token = ""
|
111 |
t = 0
|
112 |
for response in stream:
|
113 |
+
print(f"IN_{t}")
|
114 |
if (
|
115 |
(response.token.text in ["Human", "-----", "Question:"] and previous_token in ["\n", "-----"])
|
116 |
or response.token.text in ["<|endoftext|>", "<|end|>", "Answer:"]
|
|
|
122 |
print(f"Out_{t} : {output}")
|
123 |
t += 1
|
124 |
previous_token = response.token.text
|
125 |
+
print("Output = "+str(output))
|
126 |
return output.strip()
|
127 |
|
128 |
|