Add CritiFusion gradio app

app.py

@@ -7,7 +7,7 @@
 import os, re, io, json, time, base64, asyncio, inspect, traceback
 from pathlib import Path
 from typing import List, Dict, Optional, Tuple, Iterable, Set
-
+import spaces
 import torch
 from PIL import Image
 import nest_asyncio
@@ -808,7 +808,13 @@ def ui_run_full(
         return base_img, final_img, enhanced_77, meta_json
     except Exception:
         return None, None, "", traceback.format_exc()
+@spaces.GPU
+def ui_run_once_gpu(*args, **kwargs):
+    return ui_run_once(*args, **kwargs)
 
+@spaces.GPU
+def ui_run_full_gpu(*args, **kwargs):
+    return ui_run_full(*args, **kwargs)
 # =========================
 # 10) Gradio UI (matches your reference fixes)
 # =========================
@@ -863,7 +869,7 @@ with gr.Blocks(title="CritiFusion (SDXL) Demo", theme=gr.themes.Soft()) as demo:
                     meta_json = gr.Code(label="Meta / Debug (JSON)", language="json")
 
             run_btn.click(
-                fn=ui_run_once,
+                fn=ui_run_once_gpu,
                 inputs=[user_prompt, seed, H, W, preset, total_steps_refine, last_k, guidance, enabled_variants, save_outputs, out_dir],
                 outputs=[gallery, meta_json],
                 api_name=False,
@@ -905,7 +911,7 @@ with gr.Blocks(title="CritiFusion (SDXL) Demo", theme=gr.themes.Soft()) as demo:
                     meta2 = gr.Code(label="Meta / Debug (JSON)", language="json")
 
             run2.click(
-                fn=ui_run_full,
+                fn=ui_run_full_gpu,
                 inputs=[p2, seed2, H2, W2, preset2, align_mode, align_score, save2, out2],
                 outputs=[base_img, final_img, enhanced, meta2],
                 api_name=False,
@@ -913,4 +919,4 @@ with gr.Blocks(title="CritiFusion (SDXL) Demo", theme=gr.themes.Soft()) as demo:
             )
 
 # IMPORTANT: share=True fixes "localhost not accessible"
-demo.queue().launch(debug=True, share=True, max_threads=1, show_api=False)
+demo.queue().launch(debug=True, show_api=False)

requirements.txt

@@ -9,3 +9,4 @@ pillow
 numpy
 together
 nest_asyncio
+spaces