alibidaran
commited on
Commit
•
e4e5dd2
1
Parent(s):
1522ef4
Update app.py
Browse files
app.py
CHANGED
@@ -27,6 +27,8 @@ label_2id={'Psoriasis': 0,
|
|
27 |
'drug reaction': 21,
|
28 |
'peptic ulcer disease': 22,
|
29 |
'diabetes': 23}
|
|
|
|
|
30 |
def detect_symptom(symptoms):
|
31 |
inputs_id=tokenizer(symptoms,padding=True,truncation=True,return_tensors="pt")
|
32 |
output=model(inputs_id['input_ids'])
|
|
|
27 |
'drug reaction': 21,
|
28 |
'peptic ulcer disease': 22,
|
29 |
'diabetes': 23}
|
30 |
+
id2_label={i:v for v,i in label_2id.items()}
|
31 |
+
|
32 |
def detect_symptom(symptoms):
|
33 |
inputs_id=tokenizer(symptoms,padding=True,truncation=True,return_tensors="pt")
|
34 |
output=model(inputs_id['input_ids'])
|