M17idd commited on
Commit
68eec95
·
verified ·
1 Parent(s): 63c1133

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -52,11 +52,13 @@ def build_pdf_index():
52
  # ----------------- تعریف LLM از Groq -----------------
53
  groq_api_key = "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"
54
 
55
- # مستقیماً از OpenAI بدون کلاس اضافه
56
- llm = OpenAI(
57
- base_url="https://api.groq.com/openai/v1",
58
- api_key="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",
59
- model_name="mixtral-8x7b-32768"
 
 
60
  )
61
 
62
  # ----------------- تعریف SimpleRetriever -----------------
 
52
  # ----------------- تعریف LLM از Groq -----------------
53
  groq_api_key = "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"
54
 
55
+ from langchain.llms import HuggingFaceEndpoint
56
+ groq_api_key = os.environ.get("GROQ_API_KEY")
57
+
58
+ # به جای OpenAI اینو بذار:
59
+ llm = HuggingFaceEndpoint(
60
+ repo_id="deepseek-ai/deepseek-llm-67b-chat",
61
+ huggingfacehub_api_token=groq_api_key
62
  )
63
 
64
  # ----------------- تعریف SimpleRetriever -----------------