alibidaran commited on
Commit
e4e5dd2
1 Parent(s): 1522ef4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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'])