suriya7 commited on
Commit
0b2b08c
1 Parent(s): 3960f65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,9 +14,9 @@ load_dotenv()
14
  Settings.llm = HuggingFaceInferenceAPI(
15
  model_name="google/gemma-1.1-7b-it",
16
  tokenizer_name="google/gemma-1.1-7b-it",
17
- context_window=3900,
18
  token=os.getenv("HF_TOKEN"),
19
- max_new_tokens=1000,
20
  generate_kwargs={"temperature": 0.1},
21
  )
22
  Settings.embed_model = HuggingFaceEmbedding(
 
14
  Settings.llm = HuggingFaceInferenceAPI(
15
  model_name="google/gemma-1.1-7b-it",
16
  tokenizer_name="google/gemma-1.1-7b-it",
17
+ context_window=3000,
18
  token=os.getenv("HF_TOKEN"),
19
+ max_new_tokens=512,
20
  generate_kwargs={"temperature": 0.1},
21
  )
22
  Settings.embed_model = HuggingFaceEmbedding(