add examples from Fashionpedia-test & supress warnings from ONNX

app.py

@@ -134,7 +134,7 @@ def draw_predictions(
     return imgs_list
 
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=20)
 def inference(image, model_name, mask_threshold, bbox_threshold):
     """
     Load the ONNX model and run inference with the provided input `image`. Visualize the predictions and save them in a
@@ -150,9 +150,14 @@ def inference(image, model_name, mask_threshold, bbox_threshold):
         repo_id="rizavelioglu/fashionfail",
         filename="facere_plus.onnx" if model_name == "facere+" else "facere_base.onnx"
     )
+    # Session options (see https://github.com/microsoft/onnxruntime/issues/14694#issuecomment-1598429295)
+    sess_options = onnxruntime.SessionOptions()
+    sess_options.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_DISABLE_ALL
     # Create an inference session.
     ort_session = onnxruntime.InferenceSession(
-        path_onnx, providers=["CUDAExecutionProvider", "CPUExecutionProvider"]
+        path_onnx,
+        providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
+        sess_options=sess_options,
     )
 
     # compute ONNX Runtime output prediction
@@ -175,7 +180,7 @@ Failure Cases in Fashion Object Detection and Segmentation</a>. <br>Upload your
 from the dropdown menu—either `Facere` or `Facere+` <br> Check out the <a 
 href="https://rizavelioglu.github.io/fashionfail/">project page</a> for more information."""
 article = r"""
-Example images are sampled from the `FashionFail-test` set, which the models did not see during training.
+Example images are sampled from the `Fashionpedia-test` and `FashionFail-test` set, which the models did not see during training.
 
 <br>**Citation** <br>If you find our work useful in your research, please consider giving a star ⭐ and 
 a citation: 
@@ -191,16 +196,25 @@ a citation:
 """
 
 examples = [
-    ["adi_103_6.jpg",  "facere", 0.5, 0.7],
-    ["adi_103_6.jpg",  "facere+", 0.5, 0.7],
-    ["adi_1201_2.jpg", "facere", 0.5, 0.7],
-    ["adi_1201_2.jpg", "facere+", 0.5, 0.7],
-    ["adi_2149_5.jpg", "facere", 0.5, 0.7],
-    ["adi_2149_5.jpg", "facere+", 0.5, 0.7],
-    ["adi_5476_3.jpg", "facere", 0.5, 0.7],
-    ["adi_5476_3.jpg", "facere+", 0.5, 0.7],
-    ["adi_5641_4.jpg", "facere", 0.5, 0.7],
-    ["adi_5641_4.jpg", "facere+", 0.5, 0.7]
+    ["examples/0a4f8205a3b58e70eec99fbbb9422d08.jpg",  "facere", 0.5, 0.7],
+    ["examples/0a72e0f76ab9b75945f5d610508f9336.jpg",  "facere", 0.5, 0.7],
+    ["examples/0a939e0e67011aecf7195c17ecb9733c.jpg",  "facere", 0.5, 0.7],
+    ["examples/adi_9086_5.jpg", "facere", 0.5, 0.5],
+    ["examples/adi_9086_5.jpg", "facere+", 0.5, 0.5],
+    ["examples/adi_9704_1.jpg", "facere", 0.5, 0.5],
+    ["examples/adi_9704_1.jpg", "facere+", 0.5, 0.5],
+    ["examples/adi_10266_5.jpg", "facere", 0.5, 0.5],
+    ["examples/adi_10266_5.jpg", "facere+", 0.5, 0.5],
+    ["examples/adi_103_6.jpg",  "facere", 0.5, 0.5],
+    ["examples/adi_103_6.jpg",  "facere+", 0.5, 0.5],
+    ["examples/adi_1201_2.jpg", "facere", 0.5, 0.7],
+    ["examples/adi_1201_2.jpg", "facere+", 0.5, 0.7],
+    ["examples/adi_2149_5.jpg", "facere", 0.5, 0.7],
+    ["examples/adi_2149_5.jpg", "facere+", 0.5, 0.7],
+    ["examples/adi_5476_3.jpg", "facere", 0.5, 0.7],
+    ["examples/adi_5476_3.jpg", "facere+", 0.5, 0.7],
+    ["examples/adi_5641_4.jpg", "facere", 0.5, 0.7],
+    ["examples/adi_5641_4.jpg", "facere+", 0.5, 0.7]
 ]
 
 demo = gr.Interface(
@@ -223,7 +237,7 @@ demo = gr.Interface(
     description=description,
     article=article,
     examples=examples,
-    cache_examples=True,
+    cache_examples=False,
     examples_per_page=6
 )