Spaces:

waddaheaven
/

multilabel-movie-genre-classifier

Sleeping

waddaheaven commited on Jul 13

Commit

8495c81

•

1 Parent(s): 252d597

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ with open("genre_types_encoded.json", "r") as fp:
 genres = list(encode_genre_types.keys())
-inf_session = rt.InferenceSession('movie-genre-classifier-quantized.onnx')
 input_name = inf_session.get_inputs()[0].name
 output_name = inf_session.get_outputs()[0].name

 genres = list(encode_genre_types.keys())
+inf_session = rt.InferenceSession('genre-classifier-quantized.onnx')
 input_name = inf_session.get_inputs()[0].name
 output_name = inf_session.get_outputs()[0].name