aroffe commited on
Commit
0847be2
·
verified ·
1 Parent(s): 86775a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def gpu_enabled() -> bool:
9
  # If cloned, fill in SPACE_ID with your own space
10
  SPACE_ID = "aroffe/comparing-diffusion-models"
11
  runtime = api.get_space_runtime(repo_id=SPACE_ID)
12
- return runtime.hardware != (SpaceHardware.CPU_BASIC || SpaceHardware.CPU_UPGRADE)
13
 
14
 
15
  def image_mod(prompt: str, model: str, image_0: gr.Image, image_1: gr.Image) -> list[gr.Image]:
 
9
  # If cloned, fill in SPACE_ID with your own space
10
  SPACE_ID = "aroffe/comparing-diffusion-models"
11
  runtime = api.get_space_runtime(repo_id=SPACE_ID)
12
+ return (runtime.hardware != SpaceHardware.CPU_BASIC and runtime.hardware != SpaceHardware.CPU_UPGRADE)
13
 
14
 
15
  def image_mod(prompt: str, model: str, image_0: gr.Image, image_1: gr.Image) -> list[gr.Image]: