Update app.py

app.py

@@ -8,7 +8,7 @@ import subprocess
 import moviepy.editor as mp
 import base64
 
-token  = AutoTokenizer.from_pretrained('ProsusAI/finbert')
+# token  = AutoTokenizer.from_pretrained('ProsusAI/finbert')
 
 inf_session = onnxruntime.InferenceSession('classifier1-quantized.onnx')
 input_name = inf_session.get_inputs()[0].name
@@ -23,7 +23,7 @@ pipe = pipeline("automatic-speech-recognition",
                     chunk_length_s=30, device=device)
 
 ### --- Text Summary --- ###
-summarizer = pipeline("summarization", model="facebook/bart-base", device=device)
+summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", device=device)
      
 
 def video_identity(video):
@@ -72,8 +72,8 @@ def classify(video_file,encoded_video):
     sum = summary(full_text)[0]['summary_text']
 
     
-    input_ids = token(sum)['input_ids'][:512]
-    logits = inf_session.run([output_name],{input_name : [input_ids]})[0]
+    # input_ids = token(sum)['input_ids'][:512]
+    logits = inf_session.run([output_name],{input_name : [sum]})[0]
     logits = torch.FloatTensor(logits)
     probs = torch.sigmoid(logits)[0]
     probs = list(probs)