yetessam commited on
Commit
1dfe865
·
verified ·
1 Parent(s): 0ae92d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -43,11 +43,12 @@ polite_guard = PoliteGuardTool()
43
  web_search = DuckDuckGoSearchTool()
44
 
45
  # Switch to Gemini if this model gets overloaded
46
- '''
 
47
  model = HfApiModel(
48
  max_tokens=2096,
49
  temperature=0.5,
50
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
51
  custom_role_conversions=None,
52
  )
53
 
@@ -60,7 +61,7 @@ model = LiteLLMModel(
60
  api_key=os.getenv("LITELLM_API_KEY")
61
  )
62
 
63
-
64
  # Import tool from Hub
65
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
66
 
 
43
  web_search = DuckDuckGoSearchTool()
44
 
45
  # Switch to Gemini if this model gets overloaded
46
+ my_id = 'Qwen/Qwen2.5-Coder-32B-Instruct'
47
+
48
  model = HfApiModel(
49
  max_tokens=2096,
50
  temperature=0.5,
51
+ model_id= my_id,
52
  custom_role_conversions=None,
53
  )
54
 
 
61
  api_key=os.getenv("LITELLM_API_KEY")
62
  )
63
 
64
+ '''
65
  # Import tool from Hub
66
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
67