Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,12 @@ model_name = os.getenv('MODEL_NAME', 'default-model-name') # Make sure to set t
|
|
| 35 |
temperature = float(os.getenv('TEMPERATURE', 0.8))
|
| 36 |
stop_token_ids = os.getenv('STOP_TOKEN_IDS', '')
|
| 37 |
host = os.getenv('HOST', '0.0.0.0')
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
|
| 40 |
# Set OpenAI's API key and API base to use vLLM's API server.
|
| 41 |
openai_api_key = "EMPTY"
|
|
|
|
| 35 |
temperature = float(os.getenv('TEMPERATURE', 0.8))
|
| 36 |
stop_token_ids = os.getenv('STOP_TOKEN_IDS', '')
|
| 37 |
host = os.getenv('HOST', '0.0.0.0')
|
| 38 |
+
port_str = os.getenv('PORT', '8001')
|
| 39 |
+
try:
|
| 40 |
+
port = int(port_str)
|
| 41 |
+
except ValueError:
|
| 42 |
+
port = 8001
|
| 43 |
+
#port = int(os.getenv('PORT', 8001))
|
| 44 |
|
| 45 |
# Set OpenAI's API key and API base to use vLLM's API server.
|
| 46 |
openai_api_key = "EMPTY"
|