Update app.py

app.py

@@ -1,4 +1,5 @@
 from transformers import pipeline
+import torch  # Import torch library
 
 model = pipeline(
     "text-generation",
@@ -9,4 +10,4 @@ model = pipeline(
 def generate_test_cases(requirement):
     prompt = f"Generate test cases for '{requirement}' in JSON format. Output only the array."
     result = model(prompt, max_length=300)[0]["generated_text"]
-    return result.strip()
+    return result.strip()