tommytracx commited on
Commit
8f46541
·
verified ·
1 Parent(s): 20a7b41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ app = Flask(__name__)
10
  logging.basicConfig(level=logging.INFO)
11
 
12
  # Configuration
13
- OLLAMA_API_URL = os.getenv('OLLAMA_API_URL', 'https://your-ollama-space.hf.space')
14
- DEFAULT_MODEL = os.getenv('DEFAULT_MODEL', 'llama2')
15
  MAX_TOKENS = int(os.getenv('MAX_TOKENS', '2048'))
16
  TEMPERATURE = float(os.getenv('TEMPERATURE', '0.7'))
17
 
 
10
  logging.basicConfig(level=logging.INFO)
11
 
12
  # Configuration
13
+ OLLAMA_API_URL = os.getenv('OLLAMA_API_URL', 'https://huggingface.co/spaces/tommytracx/ollama-api')
14
+ DEFAULT_MODEL = os.getenv('DEFAULT_MODEL','llama2,llama2:13b,llama2:70b,codellama,neural-chat,gemma-3-270m').split(',')
15
  MAX_TOKENS = int(os.getenv('MAX_TOKENS', '2048'))
16
  TEMPERATURE = float(os.getenv('TEMPERATURE', '0.7'))
17