Update app.py

app.py

@@ -76,15 +76,23 @@ 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]
-    return full_text, sum, dict(zip(classes,map(float,probs)))
+
+    final = {
+    'text':full_text,
+    'summary':sum,
+    'label':dict(zip(classes,map(float,probs)))
+}
+    return full_text, sum, dict(zip(classes,map(float,probs))), final
 
 text1 = gr.Textbox(label="Text")
 text2 =  gr.Textbox(label="Summary")
 
+
+
     
 iface = gr.Interface(fn=classify,
                      inputs=['video','text'],
-                     outputs = [text1,text2,gr.outputs.Label(num_top_classes=3)])
+                     outputs = [text1,text2,gr.outputs.Label(num_top_classes=3),final])
 iface.launch(inline=False)