Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -5,9 +5,9 @@ from PIL import Image
|
|
5 |
from tensorflow import keras
|
6 |
from tensorflow.keras.applications.resnet50 import preprocess_input
|
7 |
|
8 |
-
autoencoder = keras.models.load_model("
|
9 |
-
encoder = keras.models.load_model("
|
10 |
-
decoder = keras.models.load_model("
|
11 |
|
12 |
|
13 |
# Define the Gradio interface
|
@@ -35,7 +35,7 @@ iface = gr.Interface(
|
|
35 |
title="Denoising Autoencoder",
|
36 |
description="Upload an image and see its denoised version using a denoising autoencoder.",
|
37 |
examples=[
|
38 |
-
["
|
39 |
],
|
40 |
)
|
41 |
|
|
|
5 |
from tensorflow import keras
|
6 |
from tensorflow.keras.applications.resnet50 import preprocess_input
|
7 |
|
8 |
+
autoencoder = keras.models.load_model("./models/denoising_autoencoder_weights.h5")
|
9 |
+
encoder = keras.models.load_model("./models/encoder.h5")
|
10 |
+
decoder = keras.models.load_model("./models/decoder.h5")
|
11 |
|
12 |
|
13 |
# Define the Gradio interface
|
|
|
35 |
title="Denoising Autoencoder",
|
36 |
description="Upload an image and see its denoised version using a denoising autoencoder.",
|
37 |
examples=[
|
38 |
+
["./example.jpg"]
|
39 |
],
|
40 |
)
|
41 |
|