Upload 171 files

README.md

@@ -1,7 +1,7 @@
 ---
 title: SUPIR
 sdk: gradio
-sdk_version: "4.16.0"
+sdk_version: "5.12.0"
 app_file: app.py
 python_version: "3.10"
 hf_oauth: true

app.py

@@ -195,9 +195,75 @@ def stage1_process(
     print('<<== stage1_process')
     return LQ, gr.update(visible = True)
 
-def stage2_process(*args, request: gr.Request = None, profile: Optional[gr.OAuthProfile] = None, **kwargs):
+def stage2_process(
+    noisy_image,
+    rotation,
+    denoise_image,
+    prompt,
+    a_prompt,
+    n_prompt,
+    num_samples,
+    min_size,
+    downscale,
+    upscale,
+    edm_steps,
+    s_stage1,
+    s_stage2,
+    s_cfg,
+    randomize_seed,
+    seed,
+    s_churn,
+    s_noise,
+    color_fix_type,
+    diff_dtype,
+    ae_dtype,
+    gamma_correction,
+    linear_CFG,
+    linear_s_stage2,
+    spt_linear_CFG,
+    spt_linear_s_stage2,
+    model_select,
+    output_format,
+    allocation,
+    request: gr.Request = None,
+    profile: Optional[gr.OAuthProfile] = None,
+):
+    username = _resolved_username(profile, request)
+    print("ZeroGPU auth user: " + (f"@{username}" if username else "<none>"))
     try:
-        return restore_in_Xmin(*args, request=request, profile=profile, **kwargs)
+        return restore_in_Xmin(
+            noisy_image,
+            rotation,
+            denoise_image,
+            prompt,
+            a_prompt,
+            n_prompt,
+            num_samples,
+            min_size,
+            downscale,
+            upscale,
+            edm_steps,
+            s_stage1,
+            s_stage2,
+            s_cfg,
+            randomize_seed,
+            seed,
+            s_churn,
+            s_noise,
+            color_fix_type,
+            diff_dtype,
+            ae_dtype,
+            gamma_correction,
+            linear_CFG,
+            linear_s_stage2,
+            spt_linear_CFG,
+            spt_linear_s_stage2,
+            model_select,
+            output_format,
+            allocation,
+            request=request,
+            profile=profile,
+        )
     except Exception as e:
         message = str(e)
         print("gradio.exceptions.Error: 'No GPU is currently available for you after 60s'")
@@ -214,6 +280,11 @@ def stage2_process(*args, request: gr.Request = None, profile: Optional[gr.OAuth
                 "Steps <= 28, and smaller output size."
             )
         if "ZeroGPU quotas" in message or "Unlogged user" in message:
+            if username:
+                raise gr.Error(
+                    "ZeroGPU did not receive your logged-in session. Refresh the page and retry. "
+                    "If this persists, upgrade the Space to Gradio 5.12+."
+                )
             raise gr.Error(
                 "ZeroGPU received this request as unlogged. Click Sign in with HF, refresh the page, and try again."
             )
@@ -729,8 +800,6 @@ with gr.Blocks() as interface:
     with gr.Row():
         if hasattr(gr, "LoginButton"):
             login_button = gr.LoginButton("Sign in with HF", scale=0, min_width=180)
-            if hasattr(login_button, "activate"):
-                login_button.activate()
         else:
             gr.Markdown("Use the Hugging Face page login to authenticate.", scale=0)
         auth_status = gr.Markdown()

requirements.txt

@@ -1,4 +1,4 @@
-gradio==4.16.0
+gradio==5.12.0
 gradio_imageslider==0.0.17
 huggingface_hub==0.20.3
 spaces==0.47.0