abrar-adnan commited on
Commit
73b7bd5
1 Parent(s): 90a2c48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,12 +5,12 @@ import torch, json
5
 
6
  tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
7
 
8
- with open("genre_types_encoded-bert-base-cased.json", "r") as fp:
9
  encode_genre_types = json.load(fp)
10
 
11
  genres = list(encode_genre_types.keys())
12
 
13
- inf_session = rt.InferenceSession('game-classifier-quantized-bert-base-cased.onnx')
14
  input_name = inf_session.get_inputs()[0].name
15
  output_name = inf_session.get_outputs()[0].name
16
 
 
5
 
6
  tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
7
 
8
+ with open("genre_types_encoded.json", "r") as fp:
9
  encode_genre_types = json.load(fp)
10
 
11
  genres = list(encode_genre_types.keys())
12
 
13
+ inf_session = rt.InferenceSession('game-classifier-quantized.onnx')
14
  input_name = inf_session.get_inputs()[0].name
15
  output_name = inf_session.get_outputs()[0].name
16