Update app.py

app.py

@@ -18,9 +18,29 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto"
 )
 
+# ---------- ROUTES -------------
+
+@app.get("/")
+async def root():
+    return {
+        "message": "🚀 FastAPI MPT Model Running on Hugging Face Spaces",
+        "endpoints": ["/", "/status", "/generate"]
+    }
+
+@app.get("/status")
+async def status():
+    return {
+        "status": "ok",
+        "model": MODEL_REPO,
+        "device": str(model.device),
+        "torch_dtype": str(model.dtype)
+    }
+
+
 class InputText(BaseModel):
     text: str
 
+
 @app.post("/generate")
 async def generate_text(data: InputText):
     inputs = tokenizer(data.text, return_tensors="pt").to(model.device)