Commit
·
a62de7d
1
Parent(s):
021e350
Revert "move spaces back into the app"
Browse filesThis reverts commit 021e3505f6e3425b5bc4316b17b0a8c3b015cb05.
app.py
CHANGED
|
@@ -31,9 +31,9 @@
|
|
| 31 |
|
| 32 |
import os
|
| 33 |
os.system("pip freeze")
|
|
|
|
| 34 |
|
| 35 |
import gradio as gr
|
| 36 |
-
import spaces
|
| 37 |
import torch as torch
|
| 38 |
from diffusers import MarigoldDepthPipeline, DDIMScheduler
|
| 39 |
from gradio_dualvision import DualVisionApp
|
|
@@ -127,7 +127,6 @@ class MarigoldDepthApp(DualVisionApp):
|
|
| 127 |
"processing_res": processing_res,
|
| 128 |
}
|
| 129 |
|
| 130 |
-
@spaces.GPU()
|
| 131 |
def process(self, image_in: Image.Image, **kwargs):
|
| 132 |
ensemble_size = kwargs.get("ensemble_size", self.DEFAULT_ENSEMBLE_SIZE)
|
| 133 |
denoise_steps = kwargs.get("denoise_steps", self.DEFAULT_DENOISE_STEPS)
|
|
@@ -168,6 +167,7 @@ with MarigoldDepthApp(
|
|
| 168 |
examples_path="files",
|
| 169 |
examples_per_page=5,
|
| 170 |
squeeze_canvas=True,
|
|
|
|
| 171 |
) as demo:
|
| 172 |
demo.queue(
|
| 173 |
api_open=False,
|
|
|
|
| 31 |
|
| 32 |
import os
|
| 33 |
os.system("pip freeze")
|
| 34 |
+
import spaces
|
| 35 |
|
| 36 |
import gradio as gr
|
|
|
|
| 37 |
import torch as torch
|
| 38 |
from diffusers import MarigoldDepthPipeline, DDIMScheduler
|
| 39 |
from gradio_dualvision import DualVisionApp
|
|
|
|
| 127 |
"processing_res": processing_res,
|
| 128 |
}
|
| 129 |
|
|
|
|
| 130 |
def process(self, image_in: Image.Image, **kwargs):
|
| 131 |
ensemble_size = kwargs.get("ensemble_size", self.DEFAULT_ENSEMBLE_SIZE)
|
| 132 |
denoise_steps = kwargs.get("denoise_steps", self.DEFAULT_DENOISE_STEPS)
|
|
|
|
| 167 |
examples_path="files",
|
| 168 |
examples_per_page=5,
|
| 169 |
squeeze_canvas=True,
|
| 170 |
+
spaces_zero_gpu_enabled=True,
|
| 171 |
) as demo:
|
| 172 |
demo.queue(
|
| 173 |
api_open=False,
|