Update app.py

app.py

@@ -37,7 +37,7 @@ def make_tensor(img):
 
 
 def predictC2S(img):
-    final_transform = TF.Resize((img.size[0], img.size[1]))
+    final_transform = TF.Resize((img.size[1], img.size[0]))
     img = np.array(img)
     img = transform(img)
     img = make_tensor(img)
@@ -52,7 +52,7 @@ def predictC2S(img):
 
 
 def predictS2C(img, ref):
-    final_transform = TF.Resize((img.size[0], img.size[1]))
+    final_transform = TF.Resize((img.size[1], img.size[0]))
     img = np.array(img)
     ref = np.array(ref)
     ref = transform(ref)
@@ -77,7 +77,8 @@ def predictS2C(img, ref):
 
 example_list1 = [["./examples/img1.jpg", "./examples/ref1.jpg"],
                  ["./examples/img4.jpg", "./examples/ref4.jpg"],
-                 ["./examples/img3.jpg", "./examples/ref3.jpg"]]
+                 ["./examples/img3.jpg", "./examples/ref3.jpg"],
+                 ["./examples/img5.jpg", "./examples/ref5.jpg"]]
 example_list2 = [["./examples/sketch1.jpg"],
                  ["./examples/sketch2.jpg"],
                  ["./examples/sketch3.jpg"],