File size: 1,056 Bytes
529e19e 1d83448 9915eea 1d83448 9915eea ade3956 1d83448 529e19e aad5ec4 76d3683 aad5ec4 529e19e aad5ec4 | 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 | import gradio as gr
from sentence_transformers import SentenceTransformer, util
import pandas as pd
import torch
# تحميل الموديل من Hugging Face
model = SentenceTransformer("yazied49/NAdine3")
# تحميل البيانات
df = pd.read_csv("final_special_needs_qa.csv")
questions = df["question"].tolist()
answers = df["answer"].tolist()
question_embeddings = model.encode(questions, convert_to_tensor=True)
def get_answer(user_input):
input_embedding = model.encode(user_input, convert_to_tensor=True)
cos_scores = util.pytorch_cos_sim(input_embedding, question_embeddings)[0]
best_match_idx = torch.argmax(cos_scores).item()
return answers[best_match_idx]
iface = gr.Interface(
fn=get_answer,
inputs=gr.Textbox(lines=2, placeholder="اكتب سؤالك هنا..."),
outputs="text",
title="المساعد الطبي لذوي الاحتياجات الخاصة",
description="اسأل أي سؤال متعلق بذوي الاحتياجات الخاصة وسنحاول مساعدتك."
)
iface.launch()
|