Spaces:
Build error
Build error
Extend max duration of run on zerogpu
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ splat_pipeline = DiffusionPipeline.from_pretrained(
|
|
| 33 |
).to("cuda")
|
| 34 |
|
| 35 |
|
| 36 |
-
@spaces.GPU
|
| 37 |
def run(input_image, convert):
|
| 38 |
input_image = input_image.astype("float32") / 255.0
|
| 39 |
images = image_pipeline(
|
|
|
|
| 33 |
).to("cuda")
|
| 34 |
|
| 35 |
|
| 36 |
+
@spaces.GPU(duration=600)
|
| 37 |
def run(input_image, convert):
|
| 38 |
input_image = input_image.astype("float32") / 255.0
|
| 39 |
images = image_pipeline(
|