Update

README.md

@@ -4,7 +4,7 @@ emoji: 📈
 colorFrom: green
 colorTo: indigo
 sdk: gradio
-sdk_version: 3.36.1
+sdk_version: 4.36.1
 app_file: app.py
 pinned: false
 ---

app.py

@@ -15,8 +15,14 @@ with gr.Blocks(css="style.css") as demo:
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(label="Input", type="numpy")
-            run_button = gr.Button(value="Run")
+            run_button = gr.Button()
         with gr.Column():
-            result = gr.Image(label="Result", type="numpy", elem_id="result")
-    run_button.click(fn=model.predict, inputs=input_image, outputs=result)
-demo.queue().launch()
+            result = gr.Image(label="Result", elem_id="result")
+    run_button.click(
+        fn=model.predict,
+        inputs=input_image,
+        outputs=result,
+    )
+
+if __name__ == "__main__":
+    demo.queue(max_size=20).launch()

requirements.txt

@@ -1,4 +1,4 @@
-numpy==1.22.4
-opencv-python-headless==4.5.5.64
-torch==1.11.0
-torchvision==0.12.0
+numpy==1.26.4
+opencv-python-headless==4.10.0.82
+torch==2.0.1
+torchvision==0.15.2

style.css

@@ -1,11 +1,8 @@
 h1 {
   text-align: center;
+  display: block;
 }
 div#result {
   max-width: 600px;
   max-height: 600px;
 }
-img#visitor-badge {
-  display: block;
-  margin: auto;
-}