add filepath

app.py

@@ -60,12 +60,14 @@ class Generator(nn.Module):
 
         return out
 
-# model = Generator(3, 1, 3)
-# model.load_state_dict(torch.load('model.pth'))
-# model.eval()
+model = Generator(3, 1, 3)
+model.load_state_dict(torch.load('model.pth'))
+model.eval()
 
-def sepia(input_img):
-    print(input_img.shape, np.max(input_img), np.min(input_img))
+def predict(input_img):
+	input_img = Image.open(input_img)
+	ratio = input_img.size[1] / 256.0
+	input_img = input_img.resize((int(ratio*input_img.size[0]),256), Image.ANTIALIAS)
     sepia_filter = np.array(
         [[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]]
     )
@@ -74,6 +76,6 @@ def sepia(input_img):
     return sepia_img
 
 
-iface = gr.Interface(sepia, gr.inputs.Image(shape=(256, 256)), "image")
+iface = gr.Interface(sepia, gr.inputs.Image(type='filepath'), "image")
 
 iface.launch()