krafiq commited on
Commit
a5b1965
·
1 Parent(s): 6238866

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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