Upload app.py

app.py

@@ -3,8 +3,10 @@ from random import randint
 from all_models import models
 from externalmod import gr_Interface_load
 import asyncio
+import os
 from threading import RLock
 lock = RLock()
+HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None
 
 
 def load_fn(models):
@@ -13,7 +15,7 @@ def load_fn(models):
     for model in models:
         if model not in models_load.keys():
             try:
-                m = gr_Interface_load(f'models/{model}')
+                m = gr_Interface_load(f'models/{model}', hf_token=HF_TOKEN)
             except Exception as error:
                 print(error)
                 m = gr.Interface(lambda: None, ['text'], ['image'])
@@ -52,7 +54,7 @@ async def infer(model_str, prompt, nprompt="", height=None, width=None, steps=No
     for i in range(rand):
         noise += " "
     task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn,
-                               prompt=f'{prompt} {noise}', negative_prompt=nprompt, **kwargs))
+                               prompt=f'{prompt} {noise}', negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
     await asyncio.sleep(0)
     try:
         result = await asyncio.wait_for(task, timeout=timeout)