Daemontatox commited on
Commit
0d4a9fd
·
verified ·
1 Parent(s): 2a77127

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -107,14 +107,15 @@ retriever = db.as_retriever(
107
  # streaming=True
108
  # )
109
 
110
- repo_id = "meta-llama/Llama-3.3-70B-Instruct"
111
-
112
- llm = HuggingFaceEndpoint(
113
- repo_id=repo_id,
114
- max_new_tokens=8192,
115
- temperature=0.1,
116
- huggingfacehub_api_token=HF_TOKEN,
117
- streaming=True,
 
118
 
119
  )
120
 
 
107
  # streaming=True
108
  # )
109
 
110
+ llm = ChatOpenAI(
111
+ model="meta-llama/Llama-3.3-70B-Instruct",
112
+ temperature=0,
113
+ max_tokens=None,
114
+ timeout=None,
115
+ max_retries=2,
116
+ api_key=HF_TOKEN, # if you prefer to pass api key in directly instaed of using env vars
117
+ base_url="https://api-inference.huggingface.co/v1/",
118
+ stream=True,
119
 
120
  )
121