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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -24,10 +24,8 @@ class BasicAgent:
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
- }
31
  )
32
 
33
  # Add tools
 
24
  llm = InferenceClientModel(
25
  model_id="google/flan-t5-base",
26
  token=os.getenv("test"),
27
+ "max_new_tokens": 256,
28
+ "temperature": 0.2,
 
 
29
  )
30
 
31
  # Add tools