Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 from bs4 import BeautifulSoup
 import random
 
+# Initialize models
 classification_model = pipeline("text-classification", model="plantbert_text_classification_model", tokenizer="plantbert_text_classification_model")
 mask_model = pipeline("fill-mask", model="plantbert_fill_mask_model", tokenizer="plantbert_fill_mask_model", top_k=100)
     
@@ -149,6 +150,7 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
                 species = gr.Textbox(lines=2, label="Species", placeholder="Enter a list of comma-separated binomial names here.")
+                top_k = 
             with gr.Column():
                 text_output_1 = gr.Textbox()
                 text_output_2 = gr.Image()