Kathirsci commited on
Commit
3a63849
·
verified ·
1 Parent(s): 25afa56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -9,12 +9,10 @@ HF_TOKEN = os.getenv("HF_TOKEN")
9
  llm = HuggingFaceEndpoint(
10
  repo_id="google/gemma-1.1-7b-it",
11
  task="text-generation",
12
- model_kwargs={
13
- "max_new_tokens": 512,
14
- "top_k": 5,
15
- "temperature": 0.1,
16
- "repetition_penalty": 1.03,
17
- },
18
  huggingfacehub_api_token=HF_TOKEN
19
  )
20
  template = """
 
9
  llm = HuggingFaceEndpoint(
10
  repo_id="google/gemma-1.1-7b-it",
11
  task="text-generation",
12
+ max_new_tokens=512,
13
+ top_k=5,
14
+ temperature=0.1,
15
+ repetition_penalty=1.03,
 
 
16
  huggingfacehub_api_token=HF_TOKEN
17
  )
18
  template = """