jcleee commited on
Commit
f895639
·
verified ·
1 Parent(s): 5eb7bac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -70,8 +70,10 @@ with open("agent.json", "r") as f:
70
  model_config = agent_config["model"]["data"]
71
 
72
  # === BUILD MODEL ===
 
 
73
  model = HfApiModel(
74
- model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
75
  token=os.getenv("HF_TOKEN"),
76
  temperature=0.5,
77
  max_tokens=1024,
 
70
  model_config = agent_config["model"]["data"]
71
 
72
  # === BUILD MODEL ===
73
+ from smolagents import HfApiModel
74
+
75
  model = HfApiModel(
76
+ model_id="google/gemma-2b-it",
77
  token=os.getenv("HF_TOKEN"),
78
  temperature=0.5,
79
  max_tokens=1024,