Nicolás Larenas
commited on
Update config.py
Browse files
config.py
CHANGED
|
@@ -5,7 +5,7 @@ SYSTEM_MESSAGE = "You are a helpful assistant."
|
|
| 5 |
|
| 6 |
# Model name (update if you have access to Gemini model)
|
| 7 |
MODEL_NAME = 'gemini-pro' # Use 'gemini-pro' if you have access
|
| 8 |
-
|
| 9 |
# Default parameters for AI model generation
|
| 10 |
DEFAULT_MAX_NEW_TOKENS = 512
|
| 11 |
DEFAULT_TEMPERATURE = 0.7
|
|
|
|
| 5 |
|
| 6 |
# Model name (update if you have access to Gemini model)
|
| 7 |
MODEL_NAME = 'gemini-pro' # Use 'gemini-pro' if you have access
|
| 8 |
+
MODEL_VISION_NAME = 'gemini-pro-vision'
|
| 9 |
# Default parameters for AI model generation
|
| 10 |
DEFAULT_MAX_NEW_TOKENS = 512
|
| 11 |
DEFAULT_TEMPERATURE = 0.7
|