ramysaidagieb commited on
Commit
66cae2e
·
verified ·
1 Parent(s): 26c4c4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -4,10 +4,16 @@ import chromadb
4
  import fitz # PyMuPDF
5
  from sentence_transformers import SentenceTransformer
6
  import json
 
7
  from dspy import Example, MIPROv2, Evaluate, evaluate
8
 
9
- # إعداد نموذج اللغة
10
- dspy.settings.configure(lm=dspy.LM("mistralai/Mistral-7B-Instruct-v0.2"))
 
 
 
 
 
11
 
12
  # إعداد قاعدة Chroma
13
  client = chromadb.PersistentClient(path="./chroma_db")
@@ -86,7 +92,7 @@ with gr.Blocks() as demo:
86
  gr.Markdown("## 🧠 نظام RAG عربي باستخدام DSPy + ChromaDB + Mistral")
87
 
88
  with gr.Tab("📥 تحميل وتخزين"):
89
- pdf_input = gr.File(label="ارفع ملف PDF", type="binary") # ← هنا التعديل
90
  ingest_btn = gr.Button("إضافة إلى قاعدة البيانات")
91
  ingest_out = gr.Textbox(label="نتيجة الإضافة")
92
  ingest_btn.click(ingest, inputs=pdf_input, outputs=ingest_out)
 
4
  import fitz # PyMuPDF
5
  from sentence_transformers import SentenceTransformer
6
  import json
7
+ import os
8
  from dspy import Example, MIPROv2, Evaluate, evaluate
9
 
10
+ # احصل على التوكن من متغيرات البيئة
11
+ HF_TOKEN = os.environ["HF_TOKEN"]
12
+
13
+ # إعداد نموذج اللغة باستخدام Hugging Face Inference API
14
+ dspy.settings.configure(
15
+ lm=dspy.HFInference(model="mistralai/Mistral-7B-Instruct-v0.2", token=HF_TOKEN)
16
+ )
17
 
18
  # إعداد قاعدة Chroma
19
  client = chromadb.PersistentClient(path="./chroma_db")
 
92
  gr.Markdown("## 🧠 نظام RAG عربي باستخدام DSPy + ChromaDB + Mistral")
93
 
94
  with gr.Tab("📥 تحميل وتخزين"):
95
+ pdf_input = gr.File(label="ارفع ملف PDF", type="binary")
96
  ingest_btn = gr.Button("إضافة إلى قاعدة البيانات")
97
  ingest_out = gr.Textbox(label="نتيجة الإضافة")
98
  ingest_btn.click(ingest, inputs=pdf_input, outputs=ingest_out)