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