dis-asistano / app.py
mericka0's picture
Update app.py
4b9d480 verified
raw
history blame contribute delete
880 Bytes
from flask import Flask, request, jsonify
from transformers import pipeline
from flask_cors import CORS
app = Flask(__name__)
CORS(app) # Frontend'ten gelen istekleri kabul etmek için
qa_pipeline = pipeline("question-answering", model="deepset/tinyroberta-squad2")
@app.route("/predict", methods=["POST"])
def predict():
data = request.get_json()
question = data.get("question", "")
context = (
"Diş ağrısı, diş eti hastalıkları, çürükler, diş apseleri gibi pek çok nedenden kaynaklanabilir. "
"Düzenli diş fırçalama, diş ipi kullanımı ve hekim kontrolleri önemlidir. "
"Ciddi ağrılarda mutlaka bir diş hekimine başvurulmalıdır."
)
result = qa_pipeline(question=question, context=context)
return jsonify({"answer": result["answer"]})
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860)