bravewiki commited on
Commit
6b8a2fa
·
verified ·
1 Parent(s): 89af842

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -6,7 +6,9 @@ import numpy as np
6
  from PIL import Image
7
  from huggingface_hub import login
8
 
9
- login('HF_token')
 
 
10
 
11
  # Load the model
12
  processor = AutoProcessor.from_pretrained("google/paligemma-3b-mix-224")
 
6
  from PIL import Image
7
  from huggingface_hub import login
8
 
9
+ # Get the token from environment variables (set in Hugging Face Space)
10
+ token = os.getenv("HF_Token")
11
+ login(token)
12
 
13
  # Load the model
14
  processor = AutoProcessor.from_pretrained("google/paligemma-3b-mix-224")