tensorgirl commited on
Commit
c091ff4
·
verified ·
1 Parent(s): 211b559

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,8 +28,8 @@ vectordb = Chroma.from_documents(
28
  from langchain_community.llms import HuggingFaceHub
29
 
30
  llm = HuggingFaceHub(
31
- repo_id="google/gemma-2b-it",
32
- #repo_id = "HuggingFaceH4/zephyr-7b-beta",
33
  task="text-generation",
34
  model_kwargs={
35
  "max_new_tokens": 512,
 
28
  from langchain_community.llms import HuggingFaceHub
29
 
30
  llm = HuggingFaceHub(
31
+ #repo_id="google/gemma-2b-it",
32
+ repo_id = "google/gemma-1.1-7b-it",
33
  task="text-generation",
34
  model_kwargs={
35
  "max_new_tokens": 512,