Spaces:

Rimi98
/

Relax-Teacher

Running

Rimi98 commited on Mar 31, 2023

Commit

f5b3004

•

1 Parent(s): 1b59031

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('ProsusAI/finbert')
 inf_session = onnxruntime.InferenceSession('classifier-quantized2.onnx')
 input_name = inf_session.get_inputs()[0].name

 import moviepy.editor as mp
 import base64
+token  = AutoTokenizer.from_pretrained('distilroberta-base')
 inf_session = onnxruntime.InferenceSession('classifier-quantized2.onnx')
 input_name = inf_session.get_inputs()[0].name