shakhovak commited on
Commit
e8c8539
1 Parent(s): decf09e

correct_error

Browse files
Files changed (1) hide show
  1. retrieve_bot.py +1 -1
retrieve_bot.py CHANGED
@@ -38,7 +38,7 @@ class ChatBot:
38
  with open("data/scripts_vectors.pkl", "rb") as fp:
39
  self.vect_data = pickle.load(fp)
40
  self.scripts = pd.read_pickle("data/scripts.pkl")
41
- with open('low_score_sripts.json', 'r') as f:
42
  self.low_scoring_list = json.load(f)
43
  self.tag_model = DialogTag("distilbert-base-uncased")
44
  self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
38
  with open("data/scripts_vectors.pkl", "rb") as fp:
39
  self.vect_data = pickle.load(fp)
40
  self.scripts = pd.read_pickle("data/scripts.pkl")
41
+ with open('data/low_score_sripts.json', 'r') as f:
42
  self.low_scoring_list = json.load(f)
43
  self.tag_model = DialogTag("distilbert-base-uncased")
44
  self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")