Update app.py

app.py

@@ -141,7 +141,7 @@ message contents
 
 """
 
-def attack_sus(message):
+def attack_sus(message, max_tokens, temperature, top_p):
     output = llm_client.chat_completion(
     	model="meta-llama/Llama-3.2-11B-Vision-Instruct",
     	messages=[
@@ -154,8 +154,10 @@ def attack_sus(message):
     			"content": message
     		}
     	],
-    	max_tokens=1024,
-    	stream=False,
+        max_tokens=max_tokens,
+        stream=False,
+        temperature=temperature,
+        top_p=top_p,
     )
 
     return output.choices[0].delta.content
@@ -173,7 +175,7 @@ def respond(
     suspicious = getbool(clasif_client.predict(text=message, api_name="/predict"))
 
     if suspicious == "POSITIVE":
-        for message in attack_sus(message):
+        for message in attack_sus(message, max_tokens, temperature, top_p):
             response += token
             yield response
     elif suspicious == "NEGATIVE":
@@ -187,10 +189,11 @@ def respond(
 
         messages.append({"role": "user", "content": message})
 
-        for message in client.chat_completion(
+        for message in llm_client.chat_completion(
+            model="meta-llama/Llama-3.2-11B-Vision-Instruct",
             messages,
             max_tokens=max_tokens,
-            stream=True,
+            stream=False,
             temperature=temperature,
             top_p=top_p,
         ):