Fred808 commited on
Commit
75e825d
·
verified ·
1 Parent(s): 663422f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,9 +17,9 @@ app = FastAPI(title="Hugging Face Model Transfer Service", version="1.0.0")
17
  executor = ThreadPoolExecutor(max_workers=2)
18
 
19
  # Get Hugging Face token from environment variable
20
- HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
21
  if not HF_TOKEN:
22
- raise ValueError("HUGGINGFACE_HUB_TOKEN environment variable is not set")
23
 
24
  # Hardcoded model repository ID
25
  HARDCODED_MODEL_REPO_ID = "openai/gpt-oss-20b" # Your model
 
17
  executor = ThreadPoolExecutor(max_workers=2)
18
 
19
  # Get Hugging Face token from environment variable
20
+ HF_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
21
  if not HF_TOKEN:
22
+ raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
23
 
24
  # Hardcoded model repository ID
25
  HARDCODED_MODEL_REPO_ID = "openai/gpt-oss-20b" # Your model