Update app.py

app.py

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