Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ logging.basicConfig(level=logging.INFO)
|
|
| 12 |
# Configuration
|
| 13 |
OLLAMA_BASE_URL = os.getenv('OLLAMA_BASE_URL', 'http://localhost:11434')
|
| 14 |
MODELS_DIR = os.getenv('MODELS_DIR', '/models')
|
| 15 |
-
ALLOWED_MODELS = os.getenv('ALLOWED_MODELS', 'llama2,llama2:13b,llama2:70b,codellama,neural-chat').split(',')
|
| 16 |
|
| 17 |
class OllamaManager:
|
| 18 |
def __init__(self, base_url: str):
|
|
|
|
| 12 |
# Configuration
|
| 13 |
OLLAMA_BASE_URL = os.getenv('OLLAMA_BASE_URL', 'http://localhost:11434')
|
| 14 |
MODELS_DIR = os.getenv('MODELS_DIR', '/models')
|
| 15 |
+
ALLOWED_MODELS = os.getenv('ALLOWED_MODELS', 'llama2,llama2:13b,llama2:70b,codellama,neural-chat,gemma-3-270m').split(',')
|
| 16 |
|
| 17 |
class OllamaManager:
|
| 18 |
def __init__(self, base_url: str):
|