1MR commited on
Commit
6820eb4
·
verified ·
1 Parent(s): e601cbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def load_embedding_model():
39
  def load_llm(api_key):
40
  """Loads the LLM model for answering queries."""
41
  return HuggingFaceHub(
42
- repo_id="distributed/optimized-gpt2-1b",
43
  huggingfacehub_api_token=api_key,
44
  model_kwargs={"temperature": 0.5, "max_length": 100}
45
  )
 
39
  def load_llm(api_key):
40
  """Loads the LLM model for answering queries."""
41
  return HuggingFaceHub(
42
+ repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
43
  huggingfacehub_api_token=api_key,
44
  model_kwargs={"temperature": 0.5, "max_length": 100}
45
  )