change class to plant family

app.py

@@ -349,10 +349,10 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
                 )
         with gr.Row():
             with gr.Column(scale=1):
-                class_predicted = gr.Label(label='Class Predicted',num_top_classes=10)
+                class_predicted = gr.Label(label='Plant Family Predicted',num_top_classes=10)
             with gr.Column(scale=4):
                 with gr.Accordion("Explanations "):
-                    gr.Markdown("Computing Explanations from the model")
+                    gr.Markdown("Computing Explanations from the model for Top 5 Predicted Plant Families")
                     with gr.Column():
                         with gr.Row():
 
@@ -391,7 +391,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
                 #             closest_image_4 = gr.Image(label='Fifth Closest Image',image_mode='contain',width=200, height=200)
                 #     find_closest_btn = gr.Button("Find Closest Images")
                 with gr.Accordion('Closest Fossil Images'):
-                    gr.Markdown("Finding the closest images in the dataset")
+                    gr.Markdown("Finding 5 closest images in the dataset")
                     
                     with gr.Row():
                         closest_gallery = gr.Gallery(label="Closest Images", show_label=False,elem_id="gallery",columns=[5], rows=[1],height='auto', allow_preview=True, preview=None)
@@ -430,7 +430,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
             #labels_markdown = f"<div style='width: 100%; text-align: center;'>{labels_html}</div>"
             image_caption=[]
             for i in range(5):
-                image_caption.append((images[i],"Predicted Class "+str(i)+": "+labels[i]))
+                image_caption.append((images[i],"Predicted Plant Family "+str(i)+": "+labels[i]))
             return image_caption
 
         generate_explanations.click(fn=update_exp_outputs, inputs=[original_image,model_name,explain_method,sampling_size], outputs=[exp_gallery])

closest_sample.py

@@ -130,7 +130,7 @@ def get_diagram(embedding,top_k):
     ax.bar(sorted_classes, sorted_frequencies,color=colors)
     ax.set_xlabel('Class Label')
     ax.set_ylabel('Frequency')
-    ax.set_title('Distribution of '+str(top_k) +' Closest Sample Classes')
+    ax.set_title('Distribution of Plant Family of '+str(top_k) +' Closest Samples')
     ax.set_xticklabels(class_counts.keys(), rotation=45, ha='right')
 
     # Save the diagram to a file