Update app.py

app.py

@@ -4,25 +4,8 @@ from transformers import pipeline
 model = pipeline("text-generation", model="gpt2")
 
 def generate_test_cases(requirement):
-    prompt = f"""
-    Generate test cases for '{requirement}' in JSON format. Output only the array.
-    Example:
-    [
-        {{
-            "id": 1,
-            "title": "Valid Login",
-            "steps": ["Enter valid email", "Enter valid password", "Click login"],
-            "expected_result": "Redirect to dashboard"
-        }},
-        {{
-            "id": 2,
-            "title": "Invalid Password",
-            "steps": ["Enter valid email", "Enter wrong password", "Click login"],
-            "expected_result": "Error message displayed"
-        }}
-    ]
-    """
-    result = model(prompt, max_length=300)[0]["generated_text"]
+    prompt = f"Generate test cases for '{requirement}' in JSON format. Output only the array."
+    result = model(prompt, max_length=200)[0]["generated_text"]
     return result.strip()
 
 demo = gr.Interface(
@@ -31,7 +14,7 @@ demo = gr.Interface(
     outputs="text",
     title="Test Case Generator",
     description="Enter a requirement to generate test cases.",
-    allow_flagging=False
+    flagging_mode="never"  # Valid value for flagging_mode
 )
 
 demo.launch(server_name="0.0.0.0", server_port=7860)