Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ import subprocess
|
|
| 8 |
import moviepy.editor as mp
|
| 9 |
import base64
|
| 10 |
|
| 11 |
-
token = AutoTokenizer.from_pretrained('
|
| 12 |
|
| 13 |
inf_session = onnxruntime.InferenceSession('classifier-quantized2.onnx')
|
| 14 |
input_name = inf_session.get_inputs()[0].name
|
|
|
|
| 8 |
import moviepy.editor as mp
|
| 9 |
import base64
|
| 10 |
|
| 11 |
+
token = AutoTokenizer.from_pretrained('distilroberta-base')
|
| 12 |
|
| 13 |
inf_session = onnxruntime.InferenceSession('classifier-quantized2.onnx')
|
| 14 |
input_name = inf_session.get_inputs()[0].name
|