CanHuggie commited on
Commit
d54719e
·
verified ·
1 Parent(s): 7a239f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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
- port = int(os.getenv('PORT', 8001))
 
 
 
 
 
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"