shakhovak
commited on
Commit
•
e8c8539
1
Parent(s):
decf09e
correct_error
Browse files- 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")
|