cuda/cpu based on availability

app.py

@@ -13,6 +13,7 @@ endpoint = Endpoint()
 is_shared_ui = True if "ethansmith2000/MegaEdit" in os.environ['SPACE_ID'] else False
 modelid = "runwayml/stable-diffusion-v1-5"
 is_gpu_associated = True if torch.cuda.is_available() else False
+device = "cuda" if is_gpu_associated else "cpu"
 
 # examples = [
 #     ['square_ims/american_gothic.jpg', 'A painting of two people frowning', 'A painting of two people smiling', 0.5, 3],
@@ -68,7 +69,7 @@ Parameters explained:
 
 
 def main():
-    editpipeline = MegaEdit(modelid, device="cuda", dtype=torch.float16)
+    editpipeline = MegaEdit(modelid, device=device, dtype=torch.float16)
 
     def generate(
         input_image: Image.Image,