Update app.py

app.py

@@ -176,7 +176,7 @@ def respond(
 
     if suspicious == "POSITIVE":
         for message in attack_sus(message, max_tokens, temperature, top_p):
-            return message
+            yield message
     elif suspicious == "NEGATIVE":
         messages = [{"role": "system", "content": system_message}]
 
@@ -195,7 +195,7 @@ def respond(
                 stream=True,
                 temperature=temperature,
                 top_p=top_p):
-            return message
+            yield message
     else:
         raise Exception(f"Invalid suspicious type:{suspicious}")