L138 max token added

utils/helper_functions.py

@@ -132,7 +132,12 @@ def llama2_7b_ysa(prompt: str) -> str:
     """
 
     # Define the query payload with the prompt and any additional parameters
-    query_payload: Dict[str, Any] = {"inputs": prompt, "parameters": {}}
+    query_payload: Dict[str, Any] = {
+        "inputs": prompt,
+        "parameters": {
+            "max_new_tokens": 200
+        }
+    }
 
     # Send the query to the model and store the output response
     output = query(query_payload)