Update app.py

app.py

@@ -56,6 +56,11 @@ async def generate_text(request: TextGenerationRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+# Add a root endpoint for health checks
+@app.get("/")
+async def root():
+    return {"message": "API is running!"}
+
 # Run the app
 if __name__ == "__main__":
     import uvicorn