Update app.py

app.py

@@ -76,11 +76,13 @@ def classify(video_file,encoded_video):
     logits = inf_session.run([output_name],{input_name : [input_ids]})[0]
     logits = torch.FloatTensor(logits)
     probs = torch.sigmoid(logits)[0]
+    probs = list(probs)
+    label = classes[probs.index(max(probs))]
 
     final = {
     'text':full_text,
     'summary':sum,
-    'label':dict(zip(classes,map(float,probs)))
+    'label':label,
 }
     return final