Mauricio-100 commited on
Commit
b6ad03c
·
verified ·
1 Parent(s): 44c0a31

Create server.py

Browse files
Files changed (1) hide show
  1. server.py +23 -0
server.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, request, jsonify
2
+ from transformers import pipeline
3
+
4
+ print("🧠 Chargement du modèle Gopu Agent...")
5
+ bot = pipeline("text-generation", model="tonusername/gopu-agent")
6
+
7
+ app = Flask(__name__)
8
+
9
+ @app.route("/")
10
+ def home():
11
+ return "🤖 Gopu Agent en ligne !"
12
+
13
+ @app.route("/ask", methods=["POST"])
14
+ def ask():
15
+ data = request.get_json()
16
+ prompt = data.get("prompt", "")
17
+ if not prompt:
18
+ return jsonify({"error": "prompt manquant"}), 400
19
+ response = bot(prompt, max_new_tokens=200)[0]["generated_text"]
20
+ return jsonify({"response": response})
21
+
22
+ if __name__ == "__main__":
23
+ app.run(host="0.0.0.0", port=7860)