Update README.md

README.md

@@ -1,23 +1,38 @@
-# NeuroHealth Alzheimer MRI Space
+---
+title: NeuroHealth GGUF Space
+emoji: 🤖
+colorFrom: gray
+colorTo: green
+sdk: docker
+sdk_version: "1.0"
+python_version: "3.11"
+app_file: app.py
+pinned: false
+---
 
-This space exposes Alzheimer's MRI models saved in `backend/saved_models`.
+# NeuroHealth GGUF Space
+
+This space exposes the `qwen2-1_5b-instruct-q4_0.gguf` model in `backend/saved_models` for text generation.
 
 ## Endpoints
 
-- `GET /health` — model loading status
-- `GET /models` — available Alzheimer's MRI models
-- `POST /predict/image?model_key=<key>` — MRI image prediction
+- `GET /health` — GGUF file availability
+- `POST /generate` — text generation
 
-## Supported model keys
+## Request body example
 
-- `ad_dn121`
-- `ad_dn169`
-- `ad_dn201`
-- `ad_homogeneous`
+```json
+{
+  "prompt": "Explain the role of dopamine in Parkinson's disease.",
+  "max_tokens": 256,
+  "temperature": 0.8,
+  "top_p": 0.95
+}
+```
 
 ## Run locally with Docker
 
 ```bash
-docker build -t neurohealth-ad-mri .
-docker run --rm -p 7860:7860 neurohealth-ad-mri
+docker build -t neurohealth-gguf .
+docker run --rm -p 7860:7860 neurohealth-gguf
 ```