Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 from sentence_transformers import SentenceTransformer
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from pinecone import Pinecone
+import os
 
 # ===============================
 # 🌐 اتصال به Pinecone
@@ -23,8 +24,17 @@ embedding_model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2')
 # ===============================
 # 🧠 بارگذاری مدل MT5 برای بازنویسی
 # ===============================
-tokenizer = AutoTokenizer.from_pretrained("HooshvareLab/mt5-small-persian-qa")
-mt5_model = AutoModelForSeq2SeqLM.from_pretrained("HooshvareLab/mt5-small-persian-qa")
+# دریافت توکن از متغیر محیطی
+hf_token = os.getenv("HF_TOKEN")
+
+tokenizer = AutoTokenizer.from_pretrained(
+    "HooshvareLab/mt5-small-persian-qa", 
+    use_auth_token=hf_token
+)
+mt5_model = AutoModelForSeq2SeqLM.from_pretrained(
+    "HooshvareLab/mt5-small-persian-qa", 
+    use_auth_token=hf_token
+)
 
 # ===============================
 # 🔍 بازیابی پاسخ از Pinecone
@@ -38,13 +48,6 @@ def retrieve_answer(query, threshold=0.65, top_k=1):
     else:
         return None
 
-    if len(result['matches']) > 1:
-        second_score = result['matches'][1]['score']
-        if top_score - second_score < diff_threshold:
-            return None
-
-    return top_match['metadata'].get('answer')
-
 # ===============================
 # ✨ بازنویسی پاسخ با MT5
 # ===============================