Spaces:

Rimi98
/

Relax-Teacher

Sleeping

Rimi98 commited on Mar 31, 2023

Commit

1c30db8

•

1 Parent(s): 5561bb0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import subprocess
 import moviepy.editor as mp
 import base64
-token  = AutoTokenizer.from_pretrained('distilroberta-base')
 inf_session = onnxruntime.InferenceSession('classifier1-quantized.onnx')
 input_name = inf_session.get_inputs()[0].name

 import moviepy.editor as mp
 import base64
+token  = AutoTokenizer.from_pretrained('ProsusAI/finbert')
 inf_session = onnxruntime.InferenceSession('classifier1-quantized.onnx')
 input_name = inf_session.get_inputs()[0].name