runaksh commited on
Commit
862aa0d
·
verified ·
1 Parent(s): 7450a2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ import numpy as np
14
  loaded_model = AutoModelForSequenceClassification.from_pretrained("runaksh/Symptom-2-disease_distilBERT")
15
  loaded_tokenizer = AutoTokenizer.from_pretrained("runaksh/Symptom-2-disease_distilBERT")
16
 
17
- # Function for class prediction
18
  def predict(sample, validate=True):
19
  classifier = pipeline("text-classification", model=loaded_model, tokenizer=loaded_tokenizer)
20
  pred = classifier(sample)[0]['label']
 
14
  loaded_model = AutoModelForSequenceClassification.from_pretrained("runaksh/Symptom-2-disease_distilBERT")
15
  loaded_tokenizer = AutoTokenizer.from_pretrained("runaksh/Symptom-2-disease_distilBERT")
16
 
17
+ # Function for class prediction
18
  def predict(sample, validate=True):
19
  classifier = pipeline("text-classification", model=loaded_model, tokenizer=loaded_tokenizer)
20
  pred = classifier(sample)[0]['label']