up

app-fast.py

@@ -88,8 +88,8 @@ def generate_image(
     if seed == -1:
         seed = torch.randint(0, 1_000_000, (1,)).item()
 
-    msg = "ℹ️ This spaces currently crash because of the memory usage. Please help me fix 😅"
-    raise gr.Error(msg, duration=10)
+    # msg = "ℹ️ This spaces currently crash because of the memory usage. Please help me fix 😅"
+    # raise gr.Error(msg, duration=10)
     height, width = tuple(map(int, resolution.replace(" ", "").split("x")))
     generator = torch.Generator("cuda").manual_seed(seed)
 
@@ -125,12 +125,12 @@ with gr.Blocks(title="HiDream Image Generator Fast") as demo:
             )
 
             seed = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
-            # generate_btn = gr.Button("Generate Image", variant="primary")
-            generate_btn = gr.Button(
-                "This space currently crash because of the memory usage. Please help me fix 😅",
-                variant="primary",
-                interactive=False,
-            )
+            generate_btn = gr.Button("Generate Image", variant="primary")
+            # generate_btn = gr.Button(
+            #     "This space currently crash because of the memory usage. Please help me fix 😅",
+            #     variant="primary",
+            #     interactive=False,
+            # )
             seed_used = gr.Number(label="Seed Used", interactive=False)
 
         with gr.Column():