Kkanittha commited on
Commit
2d55982
·
verified ·
1 Parent(s): 315e778

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -20,11 +20,10 @@ class BasicAgent:
20
  def __init__(self):
21
  print("BasicAgent initialized.")
22
 
23
- # Load Hugging Face model via LangChain
24
  llm = InferenceClientModel(
25
  model_id="google/flan-t5-base",
26
- token=os.getenv("test"), # Add HF_TOKEN in your Space secrets
27
- task="text2text-generation",
28
  max_new_tokens=256,
29
  temperature=0.2,
30
  )
 
20
  def __init__(self):
21
  print("BasicAgent initialized.")
22
 
23
+ # Load model
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
  )