Update app.py

app.py

@@ -8,9 +8,9 @@ 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')
+inf_session = onnxruntime.InferenceSession('classifier-quantized2.onnx')
 input_name = inf_session.get_inputs()[0].name
 output_name = inf_session.get_outputs()[0].name
 
@@ -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 : [sum]})[0]
+    input_ids = token(sum)['input_ids'][:512]
+    logits = inf_session.run([output_name],{input_name : [input_ids]})[0]
     logits = torch.FloatTensor(logits)
     probs = torch.sigmoid(logits)[0]
     probs = list(probs)