Update Check Device

app.py

@@ -116,20 +116,21 @@ with block:
         with gr.Column():
             gallery = gr.Gallery(label="Generated images", show_label=False).style(
                 grid=[2], height="auto")
-    gr.Examples(
-        examples=[
-            ["./examples/original_iso.png", "hogwarts castle",
-             "", 50, 4, 10.0, 123123123, 0.8],
-            ["./examples/original_sword.png", "flaming sword",
-             "", 50, 4, 9.0, 1734133747, 0.8],
+    if device == "cuda":
+        gr.Examples(
+            examples=[
+                ["./examples/original_iso.png", "hogwarts castle",
+                "", 50, 4, 10.0, 123123123, 0.8],
+                ["./examples/original_sword.png", "flaming sword",
+                "", 50, 4, 9.0, 1734133747, 0.8],
 
-        ],
-        inputs=[input_image, prompt, negative_prompt, steps,
-                num_samples, scale, seed, strength],
-        outputs=[gallery],
-        fn=predict,
-        cache_examples=True,
-    )
+            ],
+            inputs=[input_image, prompt, negative_prompt, steps,
+                    num_samples, scale, seed, strength],
+            outputs=[gallery],
+            fn=predict,
+            cache_examples=True,
+        )
     run_button.click(fn=predict, inputs=[input_image, prompt, negative_prompt,
                      steps, num_samples, scale, seed, strength], outputs=[gallery])