Spaces:
Running
Running
Update app.py
Browse files
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://
|
| 14 |
-
DEFAULT_MODEL = os.getenv('DEFAULT_MODEL',
|
| 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 |
|