Update app.py

app.py

@@ -4,22 +4,18 @@ import torch
 
 app = FastAPI()
 
-# Définir le chemin correct vers ton modèle
-MODEL_PATH = "fatmata/psyboy/psybot_model"  # Remplace par le bon chemin
-
-# Charger le modèle et le tokenizer
-tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
-model = AutoModelForCausalLM.from_pretrained(MODEL_PATH)
+# Charger le modèle et le tokenizer depuis Hugging Face
+MODEL_NAME = "fatmata/psybot"
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float32)
 
 @app.get("/")
-def read_root():
-    return {"message": "Hello from PsyBot API!"}
+def home():
+    return {"message": "Bienvenue sur l'API PsyBot !"}
 
-@app.post("/generate/")
-def generate_response(prompt: str):
+@app.post("/generate")
+def generate_text(prompt: str):
     inputs = tokenizer(prompt, return_tensors="pt")
-    with torch.no_grad():
-        output = model.generate(**inputs, max_length=150)
-    
-    response = tokenizer.decode(output[0], skip_special_tokens=True)
+    outputs = model.generate(**inputs, max_length=100)
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return {"response": response}