Update app.py

app.py

@@ -25,6 +25,10 @@ logger = logging.getLogger(__name__)
 model_id = "EleutherAI/gpt-neo-125M"  # Use GPT-Neo-125M for faster performance
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
+# Set pad_token if it doesn't exist
+if tokenizer.pad_token is None:
+    tokenizer.pad_token = tokenizer.eos_token  # Use eos_token as pad_token
+
 # Load the model without quantization for CPU
 logger.info("Loading model...")
 model = AutoModelForCausalLM.from_pretrained(
@@ -71,6 +75,11 @@ async def generate_text(request: TextGenerationRequest):
         raise HTTPException(status_code=500, detail=str(e))
 
 # Add a root endpoint for health checks
-@app.get("/test")
+@app.get("/")
 async def root():
+    return {"message": "Welcome to the Text Generation API! Use /generate-text to generate text."}
+
+# Add a test endpoint
+@app.get("/test")
+async def test():
     return {"message": "API is running!"}