return [gr.update(visible = True)] * 2

gradio_demo.py

@@ -67,12 +67,17 @@ if torch.cuda.device_count() > 0:
     else:
         llava_agent = None
 
+def check_upload(input_image):
+    if input_image is None:
+        raise gr.Error("Please provide an image to restore.")
+    return [gr.update(visible = True)] * 2
+
 def rotate_anti_90(image_array):
     if image_array is None:
         raise gr.Error("Please provide an image to rotate.")
 
     image_pil = Image.fromarray(image_array)
-    image_pil = Image.Image.rotate(image_pil, -90)
+    image_pil = image_pil.rotate(90)
     image_array = np.array(image_pil)
     return image_array
 
@@ -81,7 +86,7 @@ def rotate_90(image_array):
         raise gr.Error("Please provide an image to rotate.")
 
     image_pil = Image.fromarray(image_array)
-    image_pil = image_pil.rotate(90)
+    image_pil = image_pil.rotate(-90)
     image_array = np.array(image_pil)
     return image_array
 
@@ -492,7 +497,7 @@ title_html = """
     This demo can handle huge images but the process will be aborted if it lasts more than 10 min.
     Please leave a message in discussion if you encounter issues.
     
-    <p><center><a href="https://arxiv.org/abs/2401.13627">Paper</a> &emsp; <a href="http://supir.xpixel.group/">Project Page</a> &emsp; <a href="https://github.com/Fanghua-Yu/SUPIR/blob/master/assets/DemoGuide.png">How to play</a> &emsp; <a href="https://huggingface.co/blog/MonsterMMORPG/supir-sota-image-upscale-better-than-magnific-ai">Local Install Guide</a></center></p>
+    <p><center><a href="https://arxiv.org/abs/2401.13627">Paper</a> &emsp; <a href="http://supir.xpixel.group/">Project Page</a> &emsp; <a href="https://huggingface.co/blog/MonsterMMORPG/supir-sota-image-upscale-better-than-magnific-ai">Local Install Guide</a></center></p>
     """
 
 
@@ -523,9 +528,9 @@ with gr.Blocks(title="SUPIR") as interface:
     input_image = gr.Image(label="Input", show_label=True, type="numpy", height=600, elem_id="image-input")
     with gr.Row():
         with gr.Column():
-            rotate_anti_90_button = gr.Button(value="⤴ Rotate -90°", elem_id="rotate_anti_90_button")
+            rotate_anti_90_button = gr.Button(value="⤴ Rotate -90°", elem_id="rotate_anti_90_button", visible=False)
         with gr.Column():
-            rotate_90_button = gr.Button(value="⤵ Rotate +90°", elem_id="rotate_90_button")
+            rotate_90_button = gr.Button(value="⤵ Rotate +90°", elem_id="rotate_90_button", visible=False)
     with gr.Group():
         prompt = gr.Textbox(label="Image description", info="Help the AI understand what the image represents; describe as much as possible, especially the details we can't see on the original image; I advise you to write in English because other languages may not be handled", value="", placeholder="A 27 years old woman, walking, in Santiago, morning, Summer, photorealistic", lines=3)
         prompt_hint = gr.HTML("You can use a <a href='"'https://huggingface.co/spaces/MaziyarPanahi/llava-llama-3-8b'"'>LlaVa space</a> to auto-generate the description of your image.")
@@ -720,9 +725,12 @@ with gr.Blocks(title="SUPIR") as interface:
     with gr.Row():
         gr.Markdown(claim_md)
     
-    input_image.upload(fn = check, inputs = [
+    input_image.upload(fn = check_upload, inputs = [
         input_image
-    ], outputs = [], queue = False, show_progress = False)
+    ], outputs = [
+        rotate_anti_90_button,
+        rotate_90_button
+    ], queue = False, show_progress = False)
         
     rotate_anti_90_button.click(fn = rotate_anti_90, inputs = [
         input_image