Update app.py

app.py

@@ -1,12 +1,36 @@
-import os
-import subprocess
-import logging
-import re
-import random
-import string
-import requests
-import sys
-import warnings
+@njit(cache=True,nopython=True,parallel=True)
+def deps():
+    import os
+    import subprocess
+    import logging
+    import re
+    import random
+    import string
+    import requests
+    import sys
+    import warnings
+    
+    import spaces
+    import torch
+    import gradio as gr
+    import numpy as np
+    from lxml.html import fromstring
+    #from transformers import pipeline
+    from torch import multiprocessing as mp, nn
+    #from torch.multiprocessing import Pool
+    #from pathos.multiprocessing import ProcessPool as Pool
+    #from pathos.threading import ThreadPool as Pool
+    #from diffusers.pipelines.flux import FluxPipeline
+    from diffusers.utils import export_to_gif, load_image
+    from diffusers.models.modeling_utils import ModelMixin
+    from huggingface_hub import hf_hub_download
+    from safetensors.torch import load_file, save_file
+    from diffusers import DiffusionPipeline, AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler, DDIMScheduler, StableDiffusionXLPipeline, UNet2DConditionModel, AutoencoderKL, UNet3DConditionModel
+    #import jax
+    #import jax.numpy as jnp
+    from numba import jit,njit
+
+deps()
 
 warnings.filterwarnings("ignore")
 root = logging.getLogger()
@@ -98,31 +122,6 @@ def run(cmd, assert_success=False, capture_output=False, env=None, dry_run=False
 
     return result
 
-@njit(cache=True,nopython=True,parallel=True)
-def deps():
-  try:
-    import spaces
-    import torch
-    import gradio as gr
-    import numpy as np
-    from lxml.html import fromstring
-    #from transformers import pipeline
-    from torch import multiprocessing as mp, nn
-    #from torch.multiprocessing import Pool
-    #from pathos.multiprocessing import ProcessPool as Pool
-    #from pathos.threading import ThreadPool as Pool
-    #from diffusers.pipelines.flux import FluxPipeline
-    from diffusers.utils import export_to_gif, load_image
-    from diffusers.models.modeling_utils import ModelMixin
-    from huggingface_hub import hf_hub_download
-    from safetensors.torch import load_file, save_file
-    from diffusers import DiffusionPipeline, AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler, DDIMScheduler, StableDiffusionXLPipeline, UNet2DConditionModel, AutoencoderKL, UNet3DConditionModel
-    #import jax
-    #import jax.numpy as jnp
-    from numba import jit,njit
-  except Exception as e:
-    pass
-
 @njit(cache=True,nopython=True,parallel=True)
 def translate(text,lang):
     if text == None or lang == None:
@@ -319,6 +318,5 @@ def pre():
 
     mp.set_start_method("spawn", force=True)
 
-deps()
 pre()
 ui()