Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -382,7 +382,7 @@ def register(state, drawpad):
|
|
| 382 |
return state
|
| 383 |
|
| 384 |
|
| 385 |
-
@spaces.GPU
|
| 386 |
def run(state, drawpad):
|
| 387 |
model.device = torch.device('cuda')
|
| 388 |
model.reset_seed(model.generator, opt.seed)
|
|
|
|
| 382 |
return state
|
| 383 |
|
| 384 |
|
| 385 |
+
@spaces.GPU(duration=120)
|
| 386 |
def run(state, drawpad):
|
| 387 |
model.device = torch.device('cuda')
|
| 388 |
model.reset_seed(model.generator, opt.seed)
|