taesiri commited on
Commit
cbda7ef
1 Parent(s): ec76089
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -20,6 +20,11 @@ import torchvision
20
  from huggingface_hub import HfApi, login, snapshot_download
21
  from PIL import Image
22
 
 
 
 
 
 
23
  csv.field_size_limit(sys.maxsize)
24
 
25
  np.random.seed(int(time.time()))
 
20
  from huggingface_hub import HfApi, login, snapshot_download
21
  from PIL import Image
22
 
23
+
24
+ session_token = os.environ.get("SessionToken")
25
+ login(token=session_token)
26
+
27
+
28
  csv.field_size_limit(sys.maxsize)
29
 
30
  np.random.seed(int(time.time()))