kto ya commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ def generate_response(dialog, prompt_choice):
|
|
| 32 |
if '</s>' in t5_output:
|
| 33 |
t5_output = t5_output[:t5_output.find('</s>')].strip()
|
| 34 |
t5_output = t5_output.replace('<extra_id_0>', '').strip()
|
| 35 |
-
t5_output = t5_output.split('Собеседник')[0].strip()
|
| 36 |
|
| 37 |
return t5_output
|
| 38 |
|
|
@@ -46,3 +46,5 @@ iface = gr.Interface(fn=generate_response,
|
|
| 46 |
|
| 47 |
# Launch the interface
|
| 48 |
iface.launch()
|
|
|
|
|
|
|
|
|
| 32 |
if '</s>' in t5_output:
|
| 33 |
t5_output = t5_output[:t5_output.find('</s>')].strip()
|
| 34 |
t5_output = t5_output.replace('<extra_id_0>', '').strip()
|
| 35 |
+
t5_output = t5_output.split('Собеседник')[0].strip()
|
| 36 |
|
| 37 |
return t5_output
|
| 38 |
|
|
|
|
| 46 |
|
| 47 |
# Launch the interface
|
| 48 |
iface.launch()
|
| 49 |
+
|
| 50 |
+
|