egorshipix commited on
Commit
aeb9ee6
1 Parent(s): a1fcc37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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
+