Update app.py

app.py

@@ -24,7 +24,7 @@ def return_habitat_image(habitat_label):
     else:
         image_url = "https://www.salonlfc.com/wp-content/uploads/2018/01/image-not-found-scaled-1150x647.png"
     #image_url = "https://www.commissionoceanindien.org/wp-content/uploads/2018/07/plantnet.jpg"  # While we don't have the rights
-    #image_url = "https://files.ibot.cas.cz/cevs/images/syntaxa/large/Rorippo-Phalaridetum_arundinaceae2.jpg"  # 800-600 for Q51
+    image_url = "https://files.ibot.cas.cz/cevs/images/syntaxa/large/Rorippo-Phalaridetum_arundinaceae2.jpg"  # 800-600 for Q51
     image = gr.Image(value=image_url)
     return image
 
@@ -42,7 +42,7 @@ def return_species_image(species):
     else:
         image_url = "https://www.salonlfc.com/wp-content/uploads/2018/01/image-not-found-scaled-1150x647.png"
     #image_url = "https://www.commissionoceanindien.org/wp-content/uploads/2018/07/plantnet.jpg"  # While we don't have the rights
-    #image_url = "https://files.ibot.cas.cz/cevs/images/taxa/large/Eryngium_maritimum18.jpg"  # 1600-1200 for Q51 for eryngium maritimum
+    image_url = "https://files.ibot.cas.cz/cevs/images/taxa/large/Eryngium_maritimum18.jpg"  # 1600-1200 for Q51 for eryngium maritimum
     image = gr.Image(value=image_url)
     return image
 
@@ -74,8 +74,10 @@ def classification(text, k):
     habitat_labels = [res['label'] for res in result[0][:k]]
     if k == 1:
         text = f"This vegetation plot probably belongs to the habitat {habitat_labels[0]}."
-    else:
+    elif k == 2:
         text = f"This vegetation plot probably belongs to the habitat {', '.join(habitat_labels[:-1])} or {habitat_labels[-1]}."
+    else:
+        text = f"This vegetation plot probably belongs to the habitat {', '.join(habitat_labels[:-1])}, or {habitat_labels[-1]}."
     habitat_name = eunis_habitats[eunis_habitats['EUNIS 2020 code'] == habitat_labels[0]]['EUNIS-2021 habitat name'].values[0]
     text += f"\nThe most likely habitat is '{habitat_name}'."
     text += f"\nSee an image of this habitat (i.e., {habitat_labels[0]}) below."
@@ -122,8 +124,10 @@ def masking(text, k):
     
     if k == 1:
         text = f"The most likely missing species is {best_predictions[0]} (position {best_positions[0]})."
-    else:
+    elif k == 2:
         text = f"The most likely missing species are {', '.join(best_predictions[:-1])} and {best_predictions[-1]} (positions {', '.join(map(str, best_positions[:-1]))} and {best_positions[-1]})."
+    else:
+        text = f"The most likely missing species are {', '.join(best_predictions[:-1])}, and {best_predictions[-1]} (positions {', '.join(map(str, best_positions[:-1]))}, and {best_positions[-1]})."
     text += f"\nThe completed vegetation plot is '{best_sentence}'."
     text += f"\nSee an image of the most likely species (i.e., {best_predictions[0]}) below."
     image = return_species_image(best_predictions[0])