Update app_demo.py
Browse files- app_demo.py +2 -1
app_demo.py
CHANGED
|
@@ -7,7 +7,8 @@ import gradio as gr
|
|
| 7 |
import numpy as np
|
| 8 |
import PIL.Image
|
| 9 |
import torch
|
| 10 |
-
from diffusers import DiffusionPipeline
|
|
|
|
| 11 |
from tqdm import tqdm
|
| 12 |
from safetensors.torch import load_file
|
| 13 |
from concurrent.futures import ThreadPoolExecutor
|
|
|
|
| 7 |
import numpy as np
|
| 8 |
import PIL.Image
|
| 9 |
import torch
|
| 10 |
+
#from diffusers import DiffusionPipeline
|
| 11 |
+
from diffusers import StableDiffusionPipeline
|
| 12 |
from tqdm import tqdm
|
| 13 |
from safetensors.torch import load_file
|
| 14 |
from concurrent.futures import ThreadPoolExecutor
|