Update app.py

app.py

@@ -169,14 +169,16 @@ with gr.Blocks() as demo:
                                 history[-1][1] = delta
                             else:
                                 history[-1][1] += delta
+
+                            if refusal_condition(history[-1][1]):
+                                history = [['[安全拒答啟動]', '[安全拒答啟動] 請清除再開啟對話']]
+                                yield history
+                                break
                             yield history
+                    
             if history[-1][1].endswith('</s>'):
                 history[-1][1] = history[-1][1][:-4]
                 yield history
-
-            if refusal_condition(history[-1][1]):
-                history = [['[安全拒答啟動]', '[安全拒答啟動] 請清除再開啟對話']]
-                yield history
     
         print('== Record ==\nQuery: {query}\nResponse: {response}'.format(query=repr(message), response=repr(history[-1][1])))