yetessam commited on
Commit
aa945ae
·
verified ·
1 Parent(s): e1e5fc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -40,20 +40,20 @@ final_answer = FinalAnswerTool()
40
  polite_guard = PoliteGuardTool()
41
 
42
 
43
- # model = HfApiModel(
44
- # max_tokens=2096,
45
- # temperature=0.5,
46
- # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
47
  # it is possible that this model may be overloaded
48
- # custom_role_conversions=None,
49
- # )
50
-
51
- model = LiteLLMModel(
52
- model_id="gemini/gemini-2.0-flash-exp",
53
- max_tokens=2096,
54
- temperature=0.6,
55
- api_key=os.getenv("LITELLM_API_KEY")
56
- )
57
 
58
 
59
  # Import tool from Hub
 
40
  polite_guard = PoliteGuardTool()
41
 
42
 
43
+ model = HfApiModel(
44
+ max_tokens=2096,
45
+ temperature=0.5,
46
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
47
  # it is possible that this model may be overloaded
48
+ custom_role_conversions=None,
49
+ )
50
+
51
+ #model = LiteLLMModel(
52
+ # model_id="gemini/gemini-2.0-flash-exp",
53
+ # max_tokens=2096,
54
+ # temperature=0.6,
55
+ # api_key=os.getenv("LITELLM_API_KEY")
56
+ #)
57
 
58
 
59
  # Import tool from Hub