Update colorization.py

colorization.py

@@ -66,29 +66,27 @@ output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
 
 
 
-demo = gr.Interface(
+colorize = gr.Interface(
     inference, [
-        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','CodeFormer','RealESR-General-x4v3'], type="value", default='v1.4', label='version'),
+        gr.Markdown("Colorize old black & white photos"),
+        gr.Image(type="filepath", label="Input"),
+        
         gr.inputs.Number(label="Rescaling factor", default=2),
+        gr.Button("Upscale & Colorize!")
+        
         # gr.Slider(0, 100, label='Weight, only for CodeFormer. 0 for better quality, 100 for better identity', default=50)
     ], [
-        gr.outputs.Image(type="numpy", label="Output (The whole image)"),
-        gr.outputs.File(label="Download the output image")
+        gr.outputs.Image(type="numpy", label="Output (The colorized image)"),
+        gr.outputs.File(label="Download the colorized image")
     ],
     title=title,
     description=description,
     article=article,
-    # examples=[['AI-generate.jpg', 'v1.4', 2, 50], ['lincoln.jpg', 'v1.4', 2, 50], ['Blake_Lively.jpg', 'v1.4', 2, 50],
-    #           ['10045.png', 'v1.4', 2, 50]]).launch()
-    examples=[['a1.jpg', 'v1.4', 2], ['a2.jpg', 'v1.4', 2], ['a3.jpg', 'v1.4', 2],['a4.jpg', 'v1.4', 2]])
     
-demo.queue(concurrency_count=4)
-demo.launch()
-
-
+colorize.queue(concurrency_count=4)
+colorize.launch()
 
+'''
 def create_interface():
     with gr.Blocks() as enhancer:
         gr.Markdown("Colorize old black & white photos")
@@ -104,4 +102,5 @@ def run_code():
     create_interface()
 
 # The main function
-run_code()
+run_code()
+'''