Update

.gitignore

@@ -162,3 +162,4 @@ cython_debug/
 #.idea/
 .ruff_cache/
 /restored*
+/gradio_cached*

app.py

@@ -15,6 +15,10 @@ HEADER = """
         <a href="https://github.com/ZZZHANG-jx/DocRes" target="_blank" style="color: grey;">GitHub Repository</a>
     </p>
 </div>
+
+🖼️ Upload an image of a document (or choose one from examples below).  
+✔️ Choose the tasks you want to perform on the document.  
+🚀 Click "Run" and the model will enhance the document according to the selected tasks!
 """
 
 
@@ -26,9 +30,9 @@ possible_tasks = [
     "binarization",
 ]
 
-@spaces.GPU(duration=90)
-def run_tasks(image, tasks):
 
+@spaces.GPU(duration=60)
+def run_tasks(image, tasks):
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
     # load model
@@ -54,9 +58,7 @@ with gr.Blocks() as demo:
         output_image = gr.Image(label="Enhanced Image", type="numpy")
 
     button = gr.Button()
-    button.click(
-        run_tasks, inputs=[input_image, task], outputs=[output_image]
-    )
+    button.click(run_tasks, inputs=[input_image, task], outputs=[output_image])
 
     gr.Examples(
         examples=[
@@ -64,7 +66,7 @@ with gr.Blocks() as demo:
             ["input/151_in.png", ["dewarping", "deshadowing", "appearance"]],
             ["input/for_debluring.png", ["deblurring"]],
             ["input/for_appearance.png", ["appearance"]],
-            ["input/for_deshadowing.png", ["deshadowing"]],
+            ["input/for_deshadowing.jpg", ["deshadowing"]],
             ["input/for_dewarping.png", ["dewarping"]],
             ["input/for_binarization.png", ["binarization"]],
         ],
@@ -73,5 +75,5 @@ with gr.Blocks() as demo:
         fn=run_tasks,
         cache_examples="lazy",
     )
-    
+
 demo.launch()