Update app.py

app.py

@@ -1,18 +1,35 @@
 from transformers import pipeline
 
-# Use the GPT-Neo model, a larger and more capable alternative to GPT-2
-model = pipeline(
-    "text-generation",
-    model="EleutherAI/gpt-neo-1.3B",  # GPT-Neo 1.3B model
-)
+# Initialize the model pipeline
+print("Initializing model pipeline...")
+try:
+    model = pipeline(
+        "text-generation",
+        model="EleutherAI/gpt-neo-1.3B",  # Use GPT-Neo 1.3B as an example
+        torch_dtype=torch.float32  # Use float32 to avoid mixed precision issues
+    )
+    print("Model pipeline initialized successfully.")
+except Exception as e:
+    print(f"Error initializing model: {e}")
+    raise
 
 def generate_test_cases(requirement):
     # Creating a prompt for generating test cases
     prompt = f"Generate test cases for the following software requirement in JSON format: '{requirement}'. Only provide the JSON array of test cases."
-    result = model(prompt, max_length=300, num_return_sequences=1)[0]["generated_text"]
-    return result.strip()
+    
+    try:
+        print("Generating test cases...")
+        result = model(prompt, max_length=300, num_return_sequences=1)[0]["generated_text"]
+        print("Test cases generated successfully.")
+        return result.strip()
+    except Exception as e:
+        print(f"Error during generation: {e}")
+        raise
 
 # Example usage
 if __name__ == "__main__":
     requirement = "User login functionality"
-    print(generate_test_cases(requirement))
+    try:
+        print(generate_test_cases(requirement))
+    except Exception as e:
+        print(f"Error in application: {e}")