tensorgirl commited on
Commit
d793403
·
verified ·
1 Parent(s): d509340

Update app.py

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