Spaces:

runaksh
/

Symptom-2-disease_Text_Classification

Runtime error

runaksh commited on Sep 9, 2023

Commit

ecd2d5e

1 Parent(s): cfbd2d8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import numpy as np
-loaded_model = AutoModelWithLMHead.from_pretrained("runaksh/Symptom-2-disease_distilBERT")
 loaded_tokenizer = AutoTokenizer.from_pretrained("runaksh/Symptom-2-disease_distilBERT")
 # Function for class prediction

 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 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