Kkanittha commited on
Commit
c9944bf
·
verified ·
1 Parent(s): e5ecd00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ class BasicAgent:
24
  llm = InferenceClientModel(
25
  model_id="google/flan-t5-base",
26
  token=os.getenv("test"),
27
- model_kwargs={
28
  "max_new_tokens": 256,
29
  "temperature": 0.2,
30
  }
 
24
  llm = InferenceClientModel(
25
  model_id="google/flan-t5-base",
26
  token=os.getenv("test"),
27
+ generation_kwargs={
28
  "max_new_tokens": 256,
29
  "temperature": 0.2,
30
  }