Update app.py
Browse files
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 =
|
126 |
-
|
127 |
temperature=0.5,
|
128 |
-
|
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 |
|