Add cuda check

app.py

@@ -14,7 +14,7 @@ import skimage
 import skimage.measure
 from utils import *
 
-
+cuda_available = torch.cuda.is_available()
 def load_html():
     body, canvaspy = "", ""
     with open("index.html", encoding="utf8") as f:
@@ -126,22 +126,23 @@ def run_outpaint(
     state,
 ):
     base64_str = "base64"
-    # data = base64.b64decode(str(sel_buffer_str))
-    # pil = Image.open(io.BytesIO(data))
-    # sel_buffer = np.array(pil)
-    # sel_buffer[:, :, 3]=255
-    # sel_buffer[:, :, 0]=255
-    # out_pil = Image.fromarray(sel_buffer)
-    # out_buffer = io.BytesIO()
-    # out_pil.save(out_buffer, format="PNG")
-    # out_buffer.seek(0)
-    # base64_bytes = base64.b64encode(out_buffer.read())
-    # base64_str = base64_bytes.decode("ascii")
-    # return (
-    #     gr.update(label=str(state + 1), value=base64_str,),
-    #     gr.update(label="Prompt"),
-    #     state + 1,
-    # )
+    if not cuda_available:
+        data = base64.b64decode(str(sel_buffer_str))
+        pil = Image.open(io.BytesIO(data))
+        sel_buffer = np.array(pil)
+        sel_buffer[:, :, 3]=255
+        sel_buffer[:, :, 0]=255
+        out_pil = Image.fromarray(sel_buffer)
+        out_buffer = io.BytesIO()
+        out_pil.save(out_buffer, format="PNG")
+        out_buffer.seek(0)
+        base64_bytes = base64.b64encode(out_buffer.read())
+        base64_str = base64_bytes.decode("ascii")
+        return (
+            gr.update(label=str(state + 1), value=base64_str,),
+            gr.update(label="Prompt"),
+            state + 1,
+        )
     if True:
         text2img, inpaint = get_model()
         if enable_safety:
@@ -223,9 +224,9 @@ outpaint_button_js = load_js("outpaint")
 proceed_button_js = load_js("proceed")
 mode_js = load_js("mode")
 setup_button_js = load_js("setup")
-
-# def get_model(x):
-#     pass
+if not torch.cuda.is_available():
+    def get_model(x):
+        pass
 get_model(get_token())
 
 with blocks as demo: