Update app.py

app.py

@@ -26,10 +26,10 @@ MODEL_PATH='Nst_model'
 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
@@ -68,7 +68,7 @@ def gray_scaled(inp_img):
 
 def transform_mymodel(content_image,style_image):
   # Convert to float32 numpy array, add batch dimension, and normalize to range [0, 1]
-  content_image=gray_scaled(content_image)
+  #content_image=gray_scaled(content_image)
   content_image = content_image.astype(np.float32)[np.newaxis, ...] / 255.0
   style_image = style_image.astype(np.float32)[np.newaxis, ...] / 255.0