Create app.py

app.py

@@ -1,9 +1,13 @@
 from transformers import pipeline
+import os
 
-# Initialize the text generation pipeline
-model = pipeline("text-generation", model="gpt2")
+# Set cache directory to a writable location
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/cache'
+os.environ['HF_HOME'] = '/tmp/cache'
 
-def generate_text(prompt, max_length=100):
-    """Generate text from a prompt"""
-    output = model(prompt, max_length=max_length)
-    return output[0]["generated_text"]
+# Initialize model with smaller config
+model = pipeline(
+    "text-generation",
+    model="gpt2",
+    device=-1  # Force CPU usage
+)