Update app.py

app.py

@@ -1,20 +1,22 @@
+import gradio as gr
 from transformers import pipeline
 import torch
 
-# Load GPT-2 with float32 (remove torch_dtype=torch.float16)
+# Load GPT-2 with optimized parameters
 model = pipeline(
     "text-generation",
     model="gpt2",
     max_length=200,
     temperature=0.7,
     early_stopping=True,
-    torch_dtype=torch.float32  # << Use float32 instead of float16
+    torch_dtype=torch.float32  # Use float32 for CPU compatibility
 )
 
 def generate_test_cases(requirement):
     prompt = f"""
     Generate test cases for '{requirement}' in JSON format. Output only the array.
-    Example:
+    
+    Example format:
     [
         {{
             "id": 1,
@@ -30,6 +32,7 @@ def generate_test_cases(requirement):
         }}
     ]
     """
+    
     try:
         with torch.no_grad():
             result = model(prompt, max_time=10)[0]["generated_text"]
@@ -37,6 +40,7 @@ def generate_test_cases(requirement):
     except Exception as e:
         return f"Error: {str(e)}"
 
+# Create Gradio interface
 demo = gr.Interface(
     fn=generate_test_cases,
     inputs="text",
@@ -46,4 +50,11 @@ demo = gr.Interface(
     flagging_mode="never"
 )
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+# Launch the app
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        debug=True,
+        prevent_thread_lock=True
+    )