Update app.py

app.py

@@ -89,6 +89,14 @@ def predict_wrapper(image, slider_threshold=0.5):
 title = "Victor - DinoVd'eau image classification"
 model_link = "https://huggingface.co/" + checkpoint_name
 description = f"This application showcases the capability of artificial intelligence-based systems to identify objects within underwater images. To utilize it, you can either upload your own image or select one of the provided examples for analysis.\nFor predictions, we use this [open-source model]({model_link})"
+description = ("This application showcases the capability of artificial intelligence-based systems "
+               "to identify objects within underwater images. To utilize it, you can either upload "
+               "your own image or select one of the provided examples for analysis. "
+               "\nFor predictions, we use this [open-source model](model_link)")
+# New subtitle message to be added
+subtitle = "Note: the model runs on CPU, so it may take a while to run the model."
+# Combine description and subtitle
+full_description = f"{description}\n\n{subtitle}"
 
 iface = gr.Interface(
     fn=predict_wrapper,
@@ -98,7 +106,7 @@ iface = gr.Interface(
         gr.components.Label(label="Slider Threshold Predictions")
     ],
     title=title, 
-    description=description,
+    description=full_description,
     examples=[["session_GOPR0106.JPG"], 
               ["session_2021_08_30_Mayotte_10_image_00066.jpg"], 
               ["session_2018_11_17_kite_Le_Morne_Manawa_G0065777.JPG"],