WarpWingHF commited on
Commit
234c728
1 Parent(s): 64e91a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ tf.config.threading.set_intra_op_parallelism_threads(8)
17
 
18
  def load_img(image):
19
  """Load and preprocess image for style transfer"""
20
- max_dim = 1024
21
  # Convert PIL Image to tensor
22
  img = tf.convert_to_tensor(np.array(image))
23
  img = tf.image.convert_image_dtype(img, tf.float32)
 
17
 
18
  def load_img(image):
19
  """Load and preprocess image for style transfer"""
20
+ max_dim = 512
21
  # Convert PIL Image to tensor
22
  img = tf.convert_to_tensor(np.array(image))
23
  img = tf.image.convert_image_dtype(img, tf.float32)