Sa-m commited on
Commit
93326ce
1 Parent(s): 363f940

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,10 +25,10 @@ MODEL_PATH='Nst_model'
25
  np.set_printoptions(suppress=True)
26
 
27
  # Load model from TF-Hub
28
- #model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')
29
 
30
  # Load the model
31
- model = tf.keras.models.load_model(MODEL_PATH)
32
 
33
  def tensor_to_image(tensor):
34
  tensor = tensor*255
 
25
  np.set_printoptions(suppress=True)
26
 
27
  # Load model from TF-Hub
28
+ model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')
29
 
30
  # Load the model
31
+ #model = tf.keras.models.load_model(MODEL_PATH)
32
 
33
  def tensor_to_image(tensor):
34
  tensor = tensor*255