Use cuda when available

app.py

@@ -45,7 +45,10 @@ class Instance:
     
     def load_model(self, model_id='j-min/IterInpaint-CLEVR'):
         pipe = StableDiffusionInpaintPipeline.from_pretrained(model_id)
-        pipe = pipe.to('cuda')
+
+        if torch.cuda.is_available():
+            pipe = pipe.to('cuda')
+            
         def dummy(images, **kwargs):
             return images, False
         pipe.safety_checker = dummy
@@ -727,10 +730,10 @@ with Blocks(
     gr.HTML(thank_desc)
 
 main.queue(concurrency_count=1, api_open=False)
-# main.launch(share=False, show_api=False, show_error=True)
-main.launch(
-    server_name="0.0.0.0",
-    share=True,
-    # server_port=7864,
-    show_api=False, show_error=True
-    )
+main.launch(share=False, show_api=False, show_error=True)
+# main.launch(
+#     server_name="0.0.0.0",
+#     share=True,
+#     # server_port=7864,
+#     show_api=False, show_error=True
+#     )