Update app.py

app.py

@@ -1,6 +1,7 @@
 import cv2
 import numpy as np
 import paddle
+import paddle.nn.functional as F
 from Generater import Generater
 import gradio as gr
 
@@ -16,10 +17,16 @@ def style_transfer(content_img,style_img):
     g_input = content_img.astype('float32') / 127.5 - 1  # 归一化
     g_input = g_input[np.newaxis, ...].transpose(0, 3, 1, 2)  # NHWC -> NCHW
     g_input = paddle.to_tensor(g_input)  # numpy -> tensor
+    h,w = g_input.shape[-2:]
+    p = max([h,w])
+    g_input = F.interpolate(g_input,scale_factor=(256/p))
 
     g_input_s = style_img.astype('float32') / 127.5 - 1  # 归一化
     g_input_s = g_input_s[np.newaxis, ...].transpose(0, 3, 1, 2)  # NHWC -> NCHW
     g_input_s = paddle.to_tensor(g_input_s)  # numpy -> tensor
+    h,w = g_input_s.shape[-2:]
+    p = max([h,w])
+    g_input_s = F.interpolate(g_input_s,scale_factor=(256/p))
 
     i = paddle.to_tensor([1.])
     g_output = generator(g_input,g_input_s,i)