Sadihsn commited on
Commit
15fbc3b
·
verified ·
1 Parent(s): 6277378

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("distilroberta-base")
7
 
8
- with open("StackOverflow_Question_Classifier/tags_types_encoded.json", "r") as fp:
9
  encode_tags_types = json.load(fp)
10
 
11
  Tags = list(encode_tags_types.keys())
12
 
13
- inf_session = rt.InferenceSession("StackOverflow_Question_Classifier/Question-classifier-quantized.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("distilroberta-base")
7
 
8
+ with open("path_files/tags_types_encoded.json", "r") as fp:
9
  encode_tags_types = json.load(fp)
10
 
11
  Tags = list(encode_tags_types.keys())
12
 
13
+ inf_session = rt.InferenceSession("path_files/Question-classifier-quantized.onnx")
14
  input_name = inf_session.get_inputs()[0].name
15
  output_name = inf_session.get_outputs()[0].name
16