ww0 commited on
Commit
19b44fb
·
verified ·
1 Parent(s): 430bc6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -194,9 +194,9 @@ def create_vectordb(data, n_retrieved_docs: int, collection_name="YouTube"):
194
 
195
  """Load LLM"""
196
 
197
- repo_id = "mistralai/Mistral-7B-Instruct-v0.1"
198
  llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"max_new_tokens": 1000})
199
-
200
 
201
 
202
 
 
194
 
195
  """Load LLM"""
196
 
197
+ repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
198
  llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"max_new_tokens": 1000})
199
+ llm.client.api_url = 'https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3'
200
 
201
 
202