Spaces:
Running
Running
Update handle_models.py
Browse files- handle_models.py +2 -0
handle_models.py
CHANGED
@@ -3,6 +3,8 @@ from config import howManyModelsToUse,num_models,max_images,inference_timeout,MA
|
|
3 |
from all_models import models
|
4 |
import asyncio
|
5 |
from externalmod import gr_Interface_load, save_image, randomize_seed
|
|
|
|
|
6 |
default_models = models[:num_models]
|
7 |
|
8 |
def load_fn(models,HF_TOKEN):
|
|
|
3 |
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):
|