charliebaby2023 commited on
Commit
63a3448
·
verified ·
1 Parent(s): 591c7b1

Update handle_models.py

Browse files
Files changed (1) hide show
  1. handle_models.py +2 -1
handle_models.py CHANGED
@@ -1,9 +1,10 @@
 
1
  from config import howManyModelsToUse,num_models,max_images,inference_timeout,MAX_SEED,thePrompt,preSetPrompt,negPreSetPrompt
2
  from all_models import models
3
  from externalmod import gr_Interface_load, save_image, randomize_seed
4
  default_models = models[:num_models]
5
 
6
- def load_fn(models):
7
  global models_load
8
  models_load = {}
9
  for model in models:
 
1
+ import gradio as gr
2
  from config import howManyModelsToUse,num_models,max_images,inference_timeout,MAX_SEED,thePrompt,preSetPrompt,negPreSetPrompt
3
  from all_models import models
4
  from externalmod import gr_Interface_load, save_image, randomize_seed
5
  default_models = models[:num_models]
6
 
7
+ def load_fn(models,HF_TOKEN):
8
  global models_load
9
  models_load = {}
10
  for model in models: