Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,8 +12,8 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
|
|
12 |
|
13 |
# Configuration
|
14 |
OLLAMA_BASE_URL = 'http://localhost:11434'
|
15 |
-
ALLOWED_MODELS = os.getenv('ALLOWED_MODELS', 'llama2,
|
16 |
-
MAX_TOKENS = int(os.getenv('MAX_TOKENS', '
|
17 |
TEMPERATURE = float(os.getenv('TEMPERATURE', '0.7'))
|
18 |
|
19 |
class OllamaManager:
|
|
|
12 |
|
13 |
# Configuration
|
14 |
OLLAMA_BASE_URL = 'http://localhost:11434'
|
15 |
+
ALLOWED_MODELS = os.getenv('ALLOWED_MODELS', 'llama2,gemma-3-270m-GGUF:Q8_0').split(',')
|
16 |
+
MAX_TOKENS = int(os.getenv('MAX_TOKENS', '4096'))
|
17 |
TEMPERATURE = float(os.getenv('TEMPERATURE', '0.7'))
|
18 |
|
19 |
class OllamaManager:
|