Update app.py

app.py

@@ -5,10 +5,10 @@ import requests
 from bs4 import BeautifulSoup
 import random
 
-classification_model = pipeline("text-classification", model="CesarLeblanc/text_classification_model")
-mask_model = pipeline("fill-mask", model="CesarLeblanc/fill_mask_model")
+classification_model = pipeline("text-classification", model="CesarLeblanc/plantbert_text_classification_model")
+mask_model = pipeline("fill-mask", model="CesarLeblanc/plantbert_fill_mask_model")
 
-dataset = load_dataset("CesarLeblanc/text_classification_dataset")
+dataset = load_dataset("CesarLeblanc/plantbert_text_classification_dataset")
 
 def return_text(habitat_label, habitat_score, confidence):
     if habitat_score*100 > confidence: