Update app.py

app.py

@@ -3,13 +3,13 @@ import torch
 import numpy as np
 import os
 from model import gen_model
-import torchvision.transforms as T
+import torchvision import transforms
 MEAN = (0.5, 0.5, 0.5,)
 STD = (0.5, 0.5, 0.5,)
 # Model
 gen,transform_gen=gen_model()
 # print(gen)
-to_img=T.ToPILImage()
+# to_img=T.ToPILImage()
 # examples=["examples/input_0.png","examples/input_9.png"]
 example_list = [["examples/" + example] for example in os.listdir("examples")]
 # example_list=['1.jpg','2.jpg']