Anton Obukhov commited on
Commit
a62de7d
·
1 Parent(s): 021e350

Revert "move spaces back into the app"

Browse files

This reverts commit 021e3505f6e3425b5bc4316b17b0a8c3b015cb05.

Files changed (1) hide show
  1. app.py +2 -2
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,