Update app.py

app.py

@@ -176,8 +176,7 @@ def respond(
 
     if suspicious == "POSITIVE":
         for message in attack_sus(message, max_tokens, temperature, top_p):
-            response += token
-            yield response
+            return message
     elif suspicious == "NEGATIVE":
         messages = [{"role": "system", "content": system_message}]
 
@@ -193,13 +192,10 @@ def respond(
                 messages,
                 model="meta-llama/Llama-3.2-11B-Vision-Instruct",
                 max_tokens=max_tokens,
-                stream=False,
+                stream=True,
                 temperature=temperature,
                 top_p=top_p):
-            token = message.choices[0].delta.content
-
-            response += token
-            yield response
+            return message
     else:
         raise Exception(f"Invalid suspicious type:{suspicious}")