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