Update app.py

app.py

@@ -14,7 +14,7 @@ import numpy as np
 loaded_model = AutoModelForSequenceClassification.from_pretrained("runaksh/Symptom-2-disease_distilBERT")
 loaded_tokenizer = AutoTokenizer.from_pretrained("runaksh/Symptom-2-disease_distilBERT")
 
-# Function for class prediction
+# Function for class prediction 
 def predict(sample, validate=True):
   classifier = pipeline("text-classification", model=loaded_model, tokenizer=loaded_tokenizer)
   pred = classifier(sample)[0]['label']