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 |
|