Update app.py

app.py

@@ -32,7 +32,7 @@ def generate_response(dialog, prompt_choice):
     if '</s>' in t5_output:
         t5_output = t5_output[:t5_output.find('</s>')].strip()
     t5_output = t5_output.replace('<extra_id_0>', '').strip()
-    t5_output = t5_output.split('Собеседник')[0].strip()
+    t5_output = t5_output.split('Собеседник')[0].strip() 
     
     return t5_output
 
@@ -46,3 +46,5 @@ iface = gr.Interface(fn=generate_response,
 
 # Launch the interface
 iface.launch()
+
+