SUBLIME

README.md

@@ -4,7 +4,7 @@ emoji: 😁
 colorFrom: yellow
 colorTo: green
 sdk: gradio
-sdk_version: 3.26.0
+sdk_version: 3.1.7
 app_file: app.py
 pinned: false
 license: apache-2.0

app.py

@@ -62,10 +62,6 @@ def inference(img, version, scale):
             img_mode = None
 
         h, w = img.shape[0:2]
-        if h > 3500 or w > 3500:
-            print('too large size')
-            return None, None
-        
         if h < 300:
             img = cv2.resize(img, (w * 2, h * 2), interpolation=cv2.INTER_LANCZOS4)
 
@@ -131,14 +127,14 @@ If you have any question, please email 📧 `xintao.wang@outlook.com` or `xintao
 """
 demo = gr.Interface(
     inference, [
-        gr.Image(type="filepath", label="Input"),
-        # gr.Radio(['v1.2', 'v1.3', 'v1.4', 'RestoreFormer', 'CodeFormer'], type="value", value='v1.4', label='version'),
-        gr.Radio(['v1.2', 'v1.3', 'v1.4', 'RestoreFormer'], type="value", value='v1.4', label='version'),
-        gr.Number(label="Rescaling factor", value=2),
-        # gr.Slider(0, 100, label='Weight, only for CodeFormer. 0 for better quality, 100 for better identity', value=50)
+        gr.inputs.Image(type="filepath", label="Input"),
+        # gr.inputs.Radio(['v1.2', 'v1.3', 'v1.4', 'RestoreFormer', 'CodeFormer'], type="value", default='v1.4', label='version'),
+        gr.inputs.Radio(['v1.2', 'v1.3', 'v1.4', 'RestoreFormer'], type="value", default='v1.4', label='version'),
+        gr.inputs.Number(label="Rescaling factor", default=2),
+        # gr.Slider(0, 100, label='Weight, only for CodeFormer. 0 for better quality, 100 for better identity', default=50)
     ], [
-        gr.Image(type="numpy", label="Output (The whole image)"),
-        gr.File(label="Download the output image")
+        gr.outputs.Image(type="numpy", label="Output (The whole image)"),
+        gr.outputs.File(label="Download the output image")
     ],
     title=title,
     description=description,
@@ -147,4 +143,5 @@ demo = gr.Interface(
     #           ['10045.png', 'v1.4', 2, 50]]).launch()
     examples=[['AI-generate.jpg', 'v1.4', 2], ['lincoln.jpg', 'v1.4', 2], ['Blake_Lively.jpg', 'v1.4', 2],
               ['10045.png', 'v1.4', 2]])
-demo.queue().launch()
+demo.queue(concurrency_count=4)
+demo.launch()