Update app.py

app.py

@@ -19,24 +19,17 @@ from cv2 import *
 # from PIL import Image, ImageOps
 #import requests
 #import tarfile
-'''
-url = "https://storage.googleapis.com/tfhub-modules/google/magenta/arbitrary-image-stylization-v1-256/2.tar.gz"
-response = requests.get(url,stream=True)
-path_input="./"
-urllib.request.urlretrieve(url, filename=path_input)
-file = tarfile.open(fileobj=response.raw, mode="r|gz")
-file.extractall(path="./nst_model")
-'''
+
 MODEL_PATH='Nst_model'
 
 # Disable scientific notation for clarity
 np.set_printoptions(suppress=True)
 
 # Load model from TF-Hub
-model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2') 
+#model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2') 
 
 # Load the model
-#model = tf.keras.models.load_model(MODEL_PATH)
+model = tf.keras.models.load_model(MODEL_PATH)
 
 def tensor_to_image(tensor):
   tensor = tensor*255