Sa-m commited on
Commit
37a22fc
1 Parent(s): d4e4b48

Update app.py

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