charliebaby2023 commited on
Commit
a83c936
·
verified ·
1 Parent(s): 40bfc4e

Update handle_models.py

Browse files
Files changed (1) hide show
  1. 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):