Update app.py

app.py

@@ -56,12 +56,13 @@ def classification(text, k):
     else:
         text = f"This vegetation plot probably belongs to the habitat type {', '.join(habitat_labels[:-1])}, or {habitat_labels[-1]}."
         text += f"\nThe most likely habitat type (i.e., {habitat_labels[0]}) is named '{habitat_name}'."
-    image_output = return_image("classification", habitat_labels[0])
-    if image_output is not None:
+    image = return_image("classification", habitat_labels[0])
+    if image is not None:
         text += "\nBelow is an example of this habitat type taken from the website FloraVEG."
     else:
         text += f"\nNo image found for this habitat type."
-    return text, image_output
+        image = gr.Image("https://img.freepik.com/premium-vector/file-folder-mascot-character-design-vector_166742-4413.jpg")
+    return text, image
 
 def masking(text, k):
     text = gbif_normalization(text)
@@ -118,7 +119,7 @@ def masking(text, k):
         text += f"\nBelow is an image of the first missing species (i.e., {best_predictions[0].capitalize()}) taken from the website FloraVEG."
     else:
         text += f"\nNo image found for the first missing species (i.e., {best_predictions[0].capitalize()})."
-
+        image = gr.Image("https://img.freepik.com/premium-vector/file-folder-mascot-character-design-vector_166742-4413.jpg")
     return text, image
 
 with gr.Blocks() as demo: