Spaces:
Runtime error
Runtime error
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):
|