Update app.py

app.py

@@ -16,7 +16,13 @@ from PIL import Image
 import numpy as np
 from model import ContentLoss, gram_matrix, StyleLoss, image_transform, get_input_optimizer,get_style_model_and_losses
 
+#Defining the Model
+cnn = models.vgg19(pretrained=True).features.eval()
 
+#Normalization
+
+cnn_normalization_mean = torch.tensor([0.485, 0.456, 0.406])
+cnn_normalization_std = torch.tensor([0.229, 0.224, 0.225])
 
 
 
@@ -125,8 +131,8 @@ example_list = [["examples/" + example] for example in os.listdir("examples")]
 demo = gr.Interface(
     fn=style_transfer,
     inputs=[
-        gr.inputs.Image(label="content Image",type='pil'),
-        gr.inputs.Image(label="style_image",type='pil')
+        gr.inputs.Image(label="content Image"),
+        gr.inputs.Image(label="style_image")
     ],
     examples=example_list,
     outputs="image",