Spaces:
Sleeping
Sleeping
File size: 966 Bytes
6807c19 e22ddd0 bbeb1f0 6807c19 e22ddd0 6807c19 e22ddd0 bbeb1f0 6807c19 bbeb1f0 6807c19 bbeb1f0 6807c19 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
from flask import Flask, request, jsonify
from gradio_client import Client
# Initialisation
app = Flask(__name__)
client = Client("Qwen/Qwen1.5-72B-Chat")
# Prompt système
system_prompt = "Tu es un assistant juridique spécialisé dans le droit du travail au Bénin. Donne des réponses claires, précises et utiles."
@app.route("/ask", methods=["POST"])
def ask_question():
data = request.get_json()
if not data or "question" not in data:
return jsonify({"error": "Champ 'question' manquant"}), 400
question = data["question"]
try:
response = client.predict(
question,
[], # pas d'historique
system_prompt,
api_name="/model_chat"
)
reponse_finale = response[1][-1][1]
return jsonify({"reponse": reponse_finale})
except Exception as e:
return jsonify({"error": str(e)}), 500
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860)
|