codys12 commited on
Commit
f473751
1 Parent(s): 70379ea
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -73,9 +73,12 @@ def generate(
73
  for text in streamer:
74
  outputs.append(text)
75
  combined_text = "".join(outputs)
76
- yield combined_text
77
  if "<<<<<<<" in combined_text:
 
 
78
  break
 
 
79
 
80
 
81
  chat_interface = gr.ChatInterface(
 
73
  for text in streamer:
74
  outputs.append(text)
75
  combined_text = "".join(outputs)
 
76
  if "<<<<<<<" in combined_text:
77
+ combined_text = combined_text.replace("<<<<<<<", "") # Remove the unwanted string
78
+ yield combined_text
79
  break
80
+ else:
81
+ yield combined_text
82
 
83
 
84
  chat_interface = gr.ChatInterface(