Update app.py
Browse files
app.py
CHANGED
|
@@ -218,6 +218,7 @@ GRADIO_EXAMPLES = [
|
|
| 218 |
for example in EXAMPLES
|
| 219 |
]
|
| 220 |
|
|
|
|
| 221 |
def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
|
| 222 |
if randomize_seed:
|
| 223 |
seed = random.randint(0, MAX_SEED)
|
|
|
|
| 218 |
for example in EXAMPLES
|
| 219 |
]
|
| 220 |
|
| 221 |
+
@spaces.GPU() # 이 데코레이터를 다시 추가
|
| 222 |
def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
|
| 223 |
if randomize_seed:
|
| 224 |
seed = random.randint(0, MAX_SEED)
|