mohammedelfeky-ai commited on
Commit
378366d
·
verified ·
1 Parent(s): f6f7755

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -10,6 +10,7 @@ from smolagents import GradioUI
10
  import tempfile
11
  import gradio as gr
12
  import os
 
13
 
14
  '''
15
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -122,12 +123,10 @@ visit_webpage=VisitWebpageTool()
122
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
123
 
124
  # Load LLM
125
- model = HfApiModel(
126
- max_tokens=2096,
127
  temperature=0.5,
128
- model_id='Qwen/Qwen3-235B-A22B', # it is possible that this model may be overloaded
129
- provider='auto', # Using 'auto' provider instead of default
130
- custom_role_conversions=None,
131
  )
132
 
133
 
 
10
  import tempfile
11
  import gradio as gr
12
  import os
13
+ from smolagents.models import GeminiModel
14
 
15
  '''
16
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
123
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
124
 
125
  # Load LLM
126
+ model = GeminiModel(
127
+ model="gemini-1.5-pro", # Using Gemini 1.5 Pro which is powerful but has free tier
128
  temperature=0.5,
129
+ max_tokens=2048,
 
 
130
  )
131
 
132