jcleee commited on
Commit
310e352
·
verified ·
1 Parent(s): f507752

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -71,18 +71,18 @@ model_config = agent_config["model"]["data"]
71
 
72
  # === BUILD MODEL ===
73
  model = HfApiModel(
74
- model_id="meta-llama/Llama-3-8B-Instruct",
75
  token=os.getenv("HF_TOKEN"),
76
  temperature=0.5,
77
  max_tokens=2048,
78
  custom_role_conversions=None,
79
  )
80
 
81
- # Set token counters manually to avoid crash
82
  model.last_input_token_count = 0
83
  model.last_output_token_count = 0
84
 
85
 
 
86
  # === IMPORT TOOL FROM HUB ===
87
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
88
 
 
71
 
72
  # === BUILD MODEL ===
73
  model = HfApiModel(
74
+ model_id="meta-llama/Llama-3-8B-Instruct-hf",
75
  token=os.getenv("HF_TOKEN"),
76
  temperature=0.5,
77
  max_tokens=2048,
78
  custom_role_conversions=None,
79
  )
80
 
 
81
  model.last_input_token_count = 0
82
  model.last_output_token_count = 0
83
 
84
 
85
+
86
  # === IMPORT TOOL FROM HUB ===
87
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
88