Update prediction.py

prediction.py

@@ -8,7 +8,7 @@ from torch import nn
 from datetime import datetime
 from db import get_database  # Assuming you have a file db.py with get_database function to connect to MongoDB
 
-# Load tokenizer and ChemBERTa
+# Load tokenizer and ChemBERTa model
 @st.cache_resource
 def load_chemberta():
     tokenizer = AutoTokenizer.from_pretrained("seyonec/ChemBERTa-zinc-base-v1")