tommytracx commited on
Commit
92601e9
·
verified ·
1 Parent(s): c5039b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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,llama2:13b,llama2:70b,codellama,neural-chat,gemma-3-270m-GGUF:Q8_0').split(',')
16
- MAX_TOKENS = int(os.getenv('MAX_TOKENS', '2048'))
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: