Update app.py
Browse files
app.py
CHANGED
@@ -172,7 +172,8 @@ def fill_shape_with_pixels(img): #img is taken by gradio as uint8
|
|
172 |
test_img = patch_stiching(flooded_image, h, w, x0, y0)
|
173 |
|
174 |
# Loading and Compiling the Model
|
175 |
-
model_path = "/content/drive/MyDrive/Pinns_Loss_file.h5"
|
|
|
176 |
model = load_model(model_path, compile = False)
|
177 |
model.compile(loss=custom_loss, optimizer=tf.keras.optimizers.AdamW(learning_rate = 0.0001), metrics=['mae', 'cosine_proximity'])
|
178 |
|
|
|
172 |
test_img = patch_stiching(flooded_image, h, w, x0, y0)
|
173 |
|
174 |
# Loading and Compiling the Model
|
175 |
+
#model_path = "/content/drive/MyDrive/Pinns_Loss_file.h5"
|
176 |
+
model_path = "krafiq/deep-neural-networks-for-navier-stokes-equations/Pinns_Loss_file.h5"
|
177 |
model = load_model(model_path, compile = False)
|
178 |
model.compile(loss=custom_loss, optimizer=tf.keras.optimizers.AdamW(learning_rate = 0.0001), metrics=['mae', 'cosine_proximity'])
|
179 |
|