BharadhwajS commited on
Commit
d9142d4
·
verified ·
1 Parent(s): ff4c483

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,10 +7,10 @@ from langchain.prompts import PromptTemplate
7
  # Initialize the chatbot
8
  HF_TOKEN = os.getenv("HF_TOKEN")
9
  llm = HuggingFaceEndpoint(
10
- #repo_id="google/gemma-7b-it",
11
  #repo_id="google/gemma-1.1-7b-it",
12
  #repo_id="google/gemma-2b",
13
- repo_id="google/gemma-7b-it",
14
  task="text-generation",
15
  max_new_tokens=512,
16
  top_k=5,
 
7
  # Initialize the chatbot
8
  HF_TOKEN = os.getenv("HF_TOKEN")
9
  llm = HuggingFaceEndpoint(
10
+ repo_id="google/gemma-2b-it",
11
  #repo_id="google/gemma-1.1-7b-it",
12
  #repo_id="google/gemma-2b",
13
+
14
  task="text-generation",
15
  max_new_tokens=512,
16
  top_k=5,