update

app.py

@@ -12,6 +12,13 @@ DESCRIPTION = """# SPRIGHT T2I
 [SPRIGHT T2I](https://spright-t2i.github.io/) is a framework to improve the spatial consistency of text-to-image models WITHOUT compromising their fidelity aspects.
 """
 
+if torch.cuda.is_available():
+    device = "cuda"
+elif torch.backends.mps.is_available():
+    device = "mps"
+else:
+    device = "cpu"
+
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = os.getenv("CACHE_EXAMPLES", "1") == "1"
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "768"))
@@ -24,7 +31,7 @@ pipe = DiffusionPipeline.from_pretrained(
     torch_dtype=torch.float16,
     use_safetensors=True,
     token=TOKEN,
-).to("cuda")
+).to(device)
 
 
 def save_image(img):

requirements.txt

@@ -1,4 +1,4 @@
 diffusers
 accelerate
-torch
-transformers
+#torch
+transformers