Update app.py

app.py

@@ -36,6 +36,7 @@ def return_habitat_image(habitat_label, habitat_score, confidence):
         image_url = "https://www.salonlfc.com/wp-content/uploads/2018/01/image-not-found-scaled-1150x647.png"
     if habitat_score*100 < confidence:
         image_url = "https://www.salonlfc.com/wp-content/uploads/2018/01/image-not-found-scaled-1150x647.png"
+    image_url = "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQoQPZxckzsiQyFh9w7Z7aJ38d23lvLQFj4QemMFjw2lvc18iQrzDYf7EzzmD7cFdfbsZU&usqp=CAU"
     image = gr.Image(value=image_url)
     return image
 
@@ -52,6 +53,7 @@ def return_species_image(species):
             image_url = "https://www.salonlfc.com/wp-content/uploads/2018/01/image-not-found-scaled-1150x647.png"
     else:
         image_url = "https://www.salonlfc.com/wp-content/uploads/2018/01/image-not-found-scaled-1150x647.png"
+    image_url = "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQoQPZxckzsiQyFh9w7Z7aJ38d23lvLQFj4QemMFjw2lvc18iQrzDYf7EzzmD7cFdfbsZU&usqp=CAU"
     image = gr.Image(value=image_url)
     return image
 
@@ -72,7 +74,7 @@ def masking(text, task):
     pred = model(masked_text, top_k=1)
     new_species = [pred[i][0]['token_str'] for i in range(len(pred))]
     new_species = ' '.join(new_species)
-    text = text + ', ' + new_species
+    text = f"The last species from this vegetation plot is probably {new_species}."
     image = return_species_image(new_species)
     return text, image