Priyanka-Balivada commited on
Commit
64a1dc5
·
verified ·
1 Parent(s): c22a0ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -44,8 +44,13 @@ def load_llm():
44
  # 'context_length': 700}
45
  # )
46
 
 
 
 
 
 
47
  llm = HuggingFaceHub(
48
- repo_id="meta-llama/Llama-2-7b-chat-hf", # Official Llama-2 model
49
  model_kwargs={"temperature": 0.5, "max_length": 500}
50
  )
51
 
 
44
  # 'context_length': 700}
45
  # )
46
 
47
+ # llm = HuggingFaceHub(
48
+ # repo_id="meta-llama/Llama-2-7b-chat-hf", # Official Llama-2 model
49
+ # model_kwargs={"temperature": 0.5, "max_length": 500}
50
+ # )
51
+
52
  llm = HuggingFaceHub(
53
+ repo_id="mistralai/Mistral-7B-Instruct", # Free and open-source alternative
54
  model_kwargs={"temperature": 0.5, "max_length": 500}
55
  )
56