Yaron Koresh commited on
Commit
e21a983
·
verified ·
1 Parent(s): 82a2612

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -32
app.py CHANGED
@@ -1,36 +1,32 @@
1
- @njit(cache=True,nopython=True,parallel=True)
2
- def deps():
3
- import os
4
- import subprocess
5
- import logging
6
- import re
7
- import random
8
- import string
9
- import requests
10
- import sys
11
- import warnings
12
 
13
- import spaces
14
- import torch
15
- import gradio as gr
16
- import numpy as np
17
- from lxml.html import fromstring
18
- #from transformers import pipeline
19
- from torch import multiprocessing as mp, nn
20
- #from torch.multiprocessing import Pool
21
- #from pathos.multiprocessing import ProcessPool as Pool
22
- #from pathos.threading import ThreadPool as Pool
23
- #from diffusers.pipelines.flux import FluxPipeline
24
- from diffusers.utils import export_to_gif, load_image
25
- from diffusers.models.modeling_utils import ModelMixin
26
- from huggingface_hub import hf_hub_download
27
- from safetensors.torch import load_file, save_file
28
- from diffusers import DiffusionPipeline, AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler, DDIMScheduler, StableDiffusionXLPipeline, UNet2DConditionModel, AutoencoderKL, UNet3DConditionModel
29
- #import jax
30
- #import jax.numpy as jnp
31
- from numba import jit,njit
32
-
33
- deps()
34
 
35
  warnings.filterwarnings("ignore")
36
  root = logging.getLogger()
 
1
+ import os
2
+ import subprocess
3
+ import logging
4
+ import re
5
+ import random
6
+ import string
7
+ import requests
8
+ import sys
9
+ import warnings
 
 
10
 
11
+ import spaces
12
+ import torch
13
+ import gradio as gr
14
+ import numpy as np
15
+ from lxml.html import fromstring
16
+ #from transformers import pipeline
17
+ from torch import multiprocessing as mp, nn
18
+ #from torch.multiprocessing import Pool
19
+ #from pathos.multiprocessing import ProcessPool as Pool
20
+ #from pathos.threading import ThreadPool as Pool
21
+ #from diffusers.pipelines.flux import FluxPipeline
22
+ from diffusers.utils import export_to_gif, load_image
23
+ from diffusers.models.modeling_utils import ModelMixin
24
+ from huggingface_hub import hf_hub_download
25
+ from safetensors.torch import load_file, save_file
26
+ from diffusers import DiffusionPipeline, AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler, DDIMScheduler, StableDiffusionXLPipeline, UNet2DConditionModel, AutoencoderKL, UNet3DConditionModel
27
+ #import jax
28
+ #import jax.numpy as jnp
29
+ from numba import jit,njit
 
 
30
 
31
  warnings.filterwarnings("ignore")
32
  root = logging.getLogger()