up

app-full.py

@@ -19,13 +19,13 @@ MODEL_CONFIGS = {
 
 # Supported image sizes
 RESOLUTION_OPTIONS: list[str] = [
-    "1024 x 1024 (Square)",
-    "768 x 1360 (Portrait)",
-    "1360 x 768 (Landscape)",
-    "880 x 1168 (Portrait)",
-    "1168 x 880 (Landscape)",
-    "1248 x 832 (Landscape)",
-    "832 x 1248 (Portrait)",
+    "1024 x 1024",
+    "768 x 1360",
+    "1360 x 768",
+    "880 x 1168",
+    "1168 x 880",
+    "1248 x 832",
+    "832 x 1248",
 ]
 
 
@@ -60,13 +60,15 @@ pipe = HiDreamImagePipeline.from_pretrained(
 pipe.transformer = transformer
 
 
-@spaces.GPU(duration=60 * 3)
+@spaces.GPU(duration=120)
 def generate_image(
     prompt: str, resolution: str, seed: int, progress=gr.Progress(track_tqdm=True)
 ) -> tuple[PIL.Image.Image, int]:
     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)
     height, width = tuple(map(int, resolution.replace(" ", "").split("x")))
     generator = torch.Generator("cuda").manual_seed(seed)
 
@@ -84,8 +86,8 @@ def generate_image(
 
 
 # Gradio UI
-with gr.Blocks(title="HiDream Image Generator") as demo:
-    gr.Markdown("## 🌈 HiDream Image Generator")
+with gr.Blocks(title="HiDream Image Generator Full") as demo:
+    gr.Markdown("## 🌈 HiDream Image Generator Full")
 
     with gr.Row():
         with gr.Column():