Update app.py

app.py

@@ -1,16 +1,24 @@
 import gradio as gr
 import os
 
-from huggingface_hub import HfApi, ModelFilter, list_liked_repos
+from huggingface_hub import HfApi, ModelFilter, list_liked_repos, SpaceHardware
 from diffusers import DiffusionPipeline
 import torch
 
+def gpu_enabled() -> bool:
+    # If cloned, fill in SPACE_ID with your own space
+    SPACE_ID = "aroffe/comparing-diffusion-models"
+    runtime = api.get_space_runtime(repo_id=SPACE_ID)
+    return runtime.hardware != (SpaceHardware.CPU_BASIC || SpaceHardware.CPU_UPGRADE)
+
+    
 def image_mod(prompt: str, model: str, image_0: gr.Image, image_1: gr.Image) -> list[gr.Image]:
+    gpu_enabled()
     images = [image_0, image_1]
     for i, diffusion_model in enumerate(model):
         pipeline = DiffusionPipeline.from_pretrained(
             pretrained_model_name_or_path=diffusion_model,
-            torch_dtype=torch.float16,
+            torch_dtype=torch.float16 if gpu_enabled() else torch.float32,
             use_safetensors=True,
             device_map="auto"
         )