File size: 622 Bytes
3197d99 fc96b33 3197d99 aef06f1 3197d99 1ba587d 3197d99 1ba587d 3197d99 d2d3314 3197d99 d2d3314 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from fastapi import FastAPI
import os
import ollama
app = FastAPI()
# Charger le modèle (vérifier si Hugging Face Spaces est prêt)
MODEL_NAME = "allenai/WildLlama-7b-user-assistant"
OLLAMA_READY = os.getenv("HF_SPACE", "false").lower() == "true"
@app.get("/")
def home():
return {"message": "API is running!", "model_ready": OLLAMA_READY}
@app.post("/chat")
def chat(prompt: str):
if not OLLAMA_READY:
return {"error": "Ollama is not available on HF Spaces"}
response = ollama.chat(model=MODEL_NAME, messages=[{"role": "user", "content": prompt}])
return {"response": response["message"]}
|