Update app.py

app.py

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