Spaces:
Runtime error
Runtime error
Commit
•
684ca13
1
Parent(s):
3081d59
spaces update
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ import random
|
|
| 8 |
import gradio as gr
|
| 9 |
import numpy as np
|
| 10 |
import PIL.Image
|
| 11 |
-
|
| 12 |
import torch
|
| 13 |
from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
|
| 14 |
from huggingface_hub import hf_hub_download
|
|
@@ -50,7 +50,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
| 50 |
return seed
|
| 51 |
|
| 52 |
|
| 53 |
-
|
| 54 |
def generate(
|
| 55 |
prompt: str,
|
| 56 |
negative_prompt: str = "",
|
|
|
|
| 8 |
import gradio as gr
|
| 9 |
import numpy as np
|
| 10 |
import PIL.Image
|
| 11 |
+
import spaces
|
| 12 |
import torch
|
| 13 |
from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
|
| 14 |
from huggingface_hub import hf_hub_download
|
|
|
|
| 50 |
return seed
|
| 51 |
|
| 52 |
|
| 53 |
+
@spaces.GPU(enable_queue=True)
|
| 54 |
def generate(
|
| 55 |
prompt: str,
|
| 56 |
negative_prompt: str = "",
|