Update server/api.py

server/api.py

@@ -1,5 +1,14 @@
 from flask import Flask, request, jsonify
 from transformers import pipeline
+from transformers import GPTNeoXForCausalLM, GPTNeoXTokenizerFast
+
+model = GPTNeoXForCausalLM.from_pretrained(
+    "Mauricio-100/agent-ai",
+    trust_remote_code=True,
+    device_map="auto"
+)
+
+tokenizer = GPTNeoXTokenizerFast.from_pretrained("Mauricio-100/agent-ai")
 
 app = Flask(__name__)
 print("🧠 Chargement du modèle Gopu Agent pour l'API...")