retry

app.py

@@ -1,6 +1,6 @@
 import os
 from huggingface_hub import hf_hub_download
-config_path=hf_hub_download(repo_id="ibm-nasa-geospatial/Prithvi-100M-sen1floods11", filename="sen1floods11_Prithvi_100M.py", token=os.environ.get("token"))
+config_path=hf_hub_download(repo_id="vrk05/Flooding_IBM", filename="floods_fine.pth", token=os.environ.get("token"))
 ckpt=hf_hub_download(repo_id="vrk05/Flooding_IBM", filename='floods_fine.pth', token=os.environ.get("token"))
 
 import argparse
@@ -33,7 +33,6 @@ import matplotlib.pyplot as plt
 
 from skimage import exposure 
 
-
 def stretch_rgb(rgb):
     
     ls_pct=1
@@ -51,7 +50,6 @@ def open_tiff(fname):
         
     return data
 
-
 def write_tiff(img_wrt, filename, metadata):
 
     """
@@ -72,7 +70,7 @@ def write_tiff(img_wrt, filename, metadata):
             dest.write(img_wrt[i, :, :], i + 1)
     
     return filename
-
+            
 
 def get_meta(fname):
     
@@ -129,6 +127,7 @@ def inference_segmentor(model, imgs, custom_test_pipeline=None):
         result = model(return_loss=False, rescale=True, **data)
     return result
 
+
 def inference_on_file(target_image, model, custom_test_pipeline):
 
     target_image = target_image.name
@@ -208,7 +207,9 @@ with gr.Blocks() as demo:
     btn.click(fn=func, inputs=inp, outputs=[out1, out2])
     
     with gr.Row():
-        gr.Examples(examples=["Spain_7370579_S2Hand.tif"],
+        gr.Examples(examples=["India_900498_S2Hand.tif",
+                              "Spain_7370579_S2Hand.tif",
+                              "USA_430764_S2Hand.tif"],
                     inputs=inp,
                     outputs=[out1, out2],
                     preprocess=preprocess_example,
@@ -216,4 +217,4 @@ with gr.Blocks() as demo:
                     cache_examples=True,
     )
 
-demo.launch() 
+demo.launch()