IamVicky111 commited on
Commit
75da653
·
verified ·
1 Parent(s): 2ccc73d

Update app.py

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