Spaces:
Running
Running
Update handle_models.py
Browse files- handle_models.py +4 -0
handle_models.py
CHANGED
@@ -4,9 +4,13 @@ from all_models import models
|
|
4 |
import asyncio
|
5 |
from externalmod import gr_Interface_load, save_image, randomize_seed
|
6 |
import os
|
|
|
|
|
7 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
8 |
default_models = models[:num_models]
|
9 |
|
|
|
|
|
10 |
def load_fn(models,HF_TOKEN):
|
11 |
global models_load
|
12 |
models_load = {}
|
|
|
4 |
import asyncio
|
5 |
from externalmod import gr_Interface_load, save_image, randomize_seed
|
6 |
import os
|
7 |
+
from threading import RLock
|
8 |
+
lock = RLock()
|
9 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
10 |
default_models = models[:num_models]
|
11 |
|
12 |
+
|
13 |
+
|
14 |
def load_fn(models,HF_TOKEN):
|
15 |
global models_load
|
16 |
models_load = {}
|