v3 to v4.

app.py

@@ -5,12 +5,11 @@ import numpy as np
 import matplotlib.pyplot as plt
 import io
 
-
-def inference(img, template, angel):
+def inference(img, template, angle):
     color_image = cv2.imread(img.name, cv2.IMREAD_COLOR)
 
     HSV_image = cv2.cvtColor(color_image, cv2.COLOR_BGR2HSV)
-    selected_harmomic_scheme = HarmonicScheme(str(template), int(angel))
+    selected_harmomic_scheme = HarmonicScheme(str(template), int(angle))
     new_HSV_image = selected_harmomic_scheme.hue_shifted(HSV_image, num_superpixels=-1)
     # Convert HSV to BGR
     result_image = cv2.cvtColor(new_HSV_image, cv2.COLOR_HSV2BGR)
@@ -30,7 +29,6 @@ def inference(img, template, angel):
     cv2.imwrite('hue.jpg', hue_plots)
     cv2.imwrite('result_image.jpg', result_image)
 
-
     return ['result_image.jpg', 'hue.jpg']
 
 title = 'Color Harmonization'
@@ -41,16 +39,16 @@ examples = [['./examples/aim_interface.png', "V", 25], ['./examples/esfahan_unsp
 
 gr.Interface(
     inference,
-    [gr.inputs.Image(type='file', label='Original Image'),
-     gr.inputs.Dropdown(["X", "Y", "T", "I", "mirror_L", "L", "V", "i"],
-                        default="X",
-                        label="Template"),
-     gr.inputs.Slider(0, 359, label="Angle")],
-    [gr.outputs.Image(type='file', label='Harmonized Image'),
-     gr.outputs.Image(type='file', label='Hue')],
+    [gr.Image(type='file', label='Original Image'),
+     gr.Dropdown(choices=["X", "Y", "T", "I", "mirror_L", "L", "V", "i"],
+                 value="X",
+                 label="Template"),
+     gr.Slider(0, 359, label="Angle")],
+    [gr.Image(type='file', label='Harmonized Image'),
+     gr.Image(type='file', label='Hue')],
     title=title,
     description=description,
     article=article,
     examples=examples,
     # css=css,
-).launch(debug=False, enable_queue=True)
+).launch(debug=False, enable_queue=True)