Hugging Face's logo

Spaces:
AIOmarRehan
/
CNN_Autoencoder_For_Image_Denoising
Running

App Files Community
AIOmarRehan commited on
Commit
4ecc63d
·
verified ·
1 Parent(s): f92b10e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from tensorflow.keras.models import load_model
9
  model = load_model("unet_model.h5", compile=False)
10
 
11
  # Load Hugging Face dataset
12
- dataset = load_dataset("AIOmarRehan/Cropped_Yale_Faces", split="train")
13
 
14
  # Preprocess function
15
  def preprocess_image(image, target_size=(192, 176)):
 
9
  model = load_model("unet_model.h5", compile=False)
10
 
11
  # Load Hugging Face dataset
12
+ dataset = load_dataset("AIOmarRehan/Cropped_Yale_Faces")
13
 
14
  # Preprocess function
15
  def preprocess_image(image, target_size=(192, 176)):