1 files changed, 3 insertions, 5 deletions
diff --git a/main.py b/main.py
index a074f71d..e5104756 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,7 @@
 import os
 import sys
 import argparse
-import cv2
+
 import tensorflow as tf
 from skimage import transform
 
@@ -55,12 +55,10 @@ def main():
    
     
     style_image = imread(ARGS.style)
-    cv2.imshow('hi1', style_image)
-    cv2.waitKey(0)
+    
     
     style_image = transform.resize(style_image, content_image.shape)
-    cv2.imshow('hi2', style_image)
-    cv2.waitKey(0)
+    
     my_model = YourModel(content_image=content_image, style_image=style_image)
     my_model.vgg16.build([1, 255, 255, 3])
     my_model.vgg16.load_weights('vgg16_imagenet.h5', by_name=True)