DigiP-AI commited on
Commit
36d1c70
·
verified ·
1 Parent(s): e6317d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -8,6 +8,9 @@ import base64
8
  import random
9
  from transforms import RGBTransform # from source code mentioned above
10
 
 
 
 
11
 
12
  loaded_model=[]
13
  for i,model in enumerate(models):
 
8
  import random
9
  from transforms import RGBTransform # from source code mentioned above
10
 
11
+ HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None # If private or gated models aren't used, ENV setting is unnecessary.
12
+ server_timeout = 100
13
+ inference_timeout = 100
14
 
15
  loaded_model=[]
16
  for i,model in enumerate(models):