Update app.py

app.py

@@ -5,7 +5,7 @@ import tensorflow as tf
 from PIL import Image
 
 def apply_random_mask(img):
-    mask_size = 100
+    mask_size = 50
     y1, x1 = np.random.randint(0, img.shape[0] - mask_size, 1)[0], np.random.randint(0, img.shape[1] - mask_size, 1)[0]
     y2, x2 = y1 + mask_size, x1 + mask_size
 
@@ -26,7 +26,7 @@ def preprocess_image(image):
     return image
 
 def predict(image):
-    model = tf.keras.models.load_model("initial_generator_model.h5")
+    model = tf.keras.models.load_model("initial_generator_model2.keras")
     processed_image = preprocess_image(image)
     input_image = np.expand_dims(processed_image, axis=0)
     output_image = model.predict(input_image)[0]