marutitecblic commited on
Commit
f8f3df1
·
verified ·
1 Parent(s): aa16163

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -10,7 +10,7 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10
 
11
  # HF_TASK = os.getenv('HF_TASK')
12
 
13
- API_TOKEN = os.getenv['API_TOKEN'] # Ensure you replace this with your actual API token
14
 
15
  # Load processor and model
16
  PROCESSOR = AutoProcessor.from_pretrained(
 
10
 
11
  # HF_TASK = os.getenv('HF_TASK')
12
 
13
+ API_TOKEN = os.getenv('API_TOKEN') # Ensure you replace this with your actual API token
14
 
15
  # Load processor and model
16
  PROCESSOR = AutoProcessor.from_pretrained(