saritha5 commited on
Commit
657d5be
1 Parent(s): c8c34b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -37,8 +37,8 @@ def classify_compliant(text):
37
  ranking = np.argsort(scores)
38
  ranking = ranking[::-1]
39
 
40
- for i in range(len(scores)):
41
- l = config.id2label[ranking[i]]
42
  #s = scores[ranking[i]]
43
  #probs[l] = np.round(float(s), 4)
44
  return l
 
37
  ranking = np.argsort(scores)
38
  ranking = ranking[::-1]
39
 
40
+
41
+ l = config.id2label[ranking[0]]
42
  #s = scores[ranking[i]]
43
  #probs[l] = np.round(float(s), 4)
44
  return l