Spaces:
Sleeping
Sleeping
rafaaa2105
commited on
Commit
•
5bde01d
1
Parent(s):
3871549
Add ZERO decorations
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ from diffusers.models import AutoencoderKL
|
|
4 |
from diffusers import StableDiffusionXLPipeline
|
5 |
import gradio as gr
|
6 |
import requests
|
|
|
7 |
|
8 |
models_list = []
|
9 |
loras_list = ["None"]
|
@@ -82,6 +83,7 @@ def load_model(model, lora="", use_lora=False):
|
|
82 |
except Exception as e:
|
83 |
return f"Error loading model {model}: {e}"
|
84 |
|
|
|
85 |
def generate_images(
|
86 |
model_name,
|
87 |
lora_name,
|
|
|
4 |
from diffusers import StableDiffusionXLPipeline
|
5 |
import gradio as gr
|
6 |
import requests
|
7 |
+
import spaces
|
8 |
|
9 |
models_list = []
|
10 |
loras_list = ["None"]
|
|
|
83 |
except Exception as e:
|
84 |
return f"Error loading model {model}: {e}"
|
85 |
|
86 |
+
@spaces.GPU
|
87 |
def generate_images(
|
88 |
model_name,
|
89 |
lora_name,
|