update

app.py

@@ -31,9 +31,16 @@ def generate_test_cases(api_info):
             f"Test cases should include:\n- Happy path\n- Negative tests\n- Edge cases"
         )
         # Tokenize the input prompt
-        inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+        inputs = tokenizer(prompt, return_tensors="pt", truncation=True, padding=True)
+        
         # Generate output from the model
-        outputs = model.generate(inputs["input_ids"], max_length=512, num_return_sequences=1, do_sample=True)
+        outputs = model.generate(
+            inputs["input_ids"],
+            attention_mask=inputs["attention_mask"],  # Provide attention mask explicitly
+            max_length=512,
+            num_return_sequences=1,
+            do_sample=True
+        )
         # Decode the generated text
         generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
         logging.info("Test cases generated successfully.")
@@ -42,6 +49,7 @@ def generate_test_cases(api_info):
         logging.error(f"Error generating test cases: {e}")
         return "An error occurred while generating test cases."
 
+
 # Process input and generate output
 def process_input(url, method, headers, payload):
     try: