staghado commited on
Commit
b38e046
·
verified ·
1 Parent(s): a608f20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -4,15 +4,11 @@ import json
4
  import requests
5
  import gradio as gr
6
 
7
- ENDPOINT = os.environ.get("VLLM_ENDPOINT") or os.getenv("VLLM_ENDPOINT")
8
- MODEL = os.environ.get("VLLM_MODEL") or os.getenv("VLLM_MODEL")
9
-
10
- print(f"Debug - Found ENDPOINT: {ENDPOINT is not None}")
11
- print(f"Debug - Found MODEL: {MODEL is not None}")
12
- print(f"Debug - All env vars: {list(os.environ.keys())}")
13
 
14
  if not ENDPOINT or not MODEL:
15
- raise ValueError("VLLM_ENDPOINT and VLLM_MODEL environment variables must be set")
16
 
17
 
18
  def respond(
 
4
  import requests
5
  import gradio as gr
6
 
7
+ ENDPOINT = os.environ.get("VLLM_ENDPOINT")
8
+ MODEL = os.environ.get("VLLM_MODEL")
 
 
 
 
9
 
10
  if not ENDPOINT or not MODEL:
11
+ raise ValueError("VLLM_ENDPOINT and VLLM_MODEL environment variables must be set. Please add them as secrets in your Space settings.")
12
 
13
 
14
  def respond(