Update app.py
Browse files
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):
|