menikev commited on
Commit
7313e9a
1 Parent(s): 69bfe84

Update prediction_sinhala.py

Browse files
Files changed (1) hide show
  1. prediction_sinhala.py +1 -3
prediction_sinhala.py CHANGED
@@ -199,9 +199,7 @@ tokenizer = TokenizerFromPreTrained(max_len, bert)
199
  batch_size = 100
200
 
201
 
202
- testing_path = path + testing_json
203
-
204
- testing_set = TextDataset(testing_path, ['text'], tokenizer)
205
  testing_loader = DataLoader(testing_set, batch_size, shuffle=False)
206
 
207
  # prepare model
 
199
  batch_size = 100
200
 
201
 
202
+ testing_set = TextDataset(new_testing_json_path, ['text'], tokenizer)
 
 
203
  testing_loader = DataLoader(testing_set, batch_size, shuffle=False)
204
 
205
  # prepare model