Kkanittha commited on
Commit
e506591
·
verified ·
1 Parent(s): 618bfef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ class BasicAgent:
15
  def __init__(self):
16
  print("BasicAgent initialized.")
17
  model = InferenceClientModel(
18
- model_id="Qwen/Qwen2.5-7B-Instruct", # or a smaller one like HuggingFaceTB/SmolLM2-1.7B-Instruct
19
  provider="hf-inference",
20
  token=os.getenv("test"),
21
  )
 
15
  def __init__(self):
16
  print("BasicAgent initialized.")
17
  model = InferenceClientModel(
18
+ model_id= "HuggingFaceTB/SmolLM2-360M-Instruct", # or a smaller one like HuggingFaceTB/SmolLM2-1.7B-Instruct
19
  provider="hf-inference",
20
  token=os.getenv("test"),
21
  )