BharadhwajS commited on
Commit
d5d69d3
·
verified ·
1 Parent(s): 6530946

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ from langchain.prompts import PromptTemplate
7
  # Initialize the chatbot
8
  HF_TOKEN = os.getenv("HF_TOKEN")
9
  llm = HuggingFaceEndpoint(
 
10
  repo_id="google/gemma-1.1-7b-it",
11
  task="text-generation",
12
  max_new_tokens=512,
 
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
  task="text-generation",
13
  max_new_tokens=512,