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 |
|