Add application file
Browse files
app.py
CHANGED
@@ -118,7 +118,7 @@ def evaluate(
|
|
118 |
)
|
119 |
s = generation_output.sequences[0]
|
120 |
output = tokenizer.decode(s)
|
121 |
-
return output.split("### Response:")[1].strip()
|
122 |
|
123 |
|
124 |
g = gr.Interface(
|
|
|
118 |
)
|
119 |
s = generation_output.sequences[0]
|
120 |
output = tokenizer.decode(s)
|
121 |
+
return output.split("### Response:")[1].strip().split("</s>")[0]
|
122 |
|
123 |
|
124 |
g = gr.Interface(
|