Commit
•
2c5b87c
1
Parent(s):
9591b5b
Fixed some style validations on chatbot.py
Browse files- chatbot.py +3 -2
chatbot.py
CHANGED
@@ -17,7 +17,8 @@ prompt.
|
|
17 |
import argparse
|
18 |
import torch
|
19 |
|
20 |
-
from transformers import AutoModelForCausalLM, AutoTokenizer, PreTrainedModel
|
|
|
21 |
from typing import Optional, Union
|
22 |
|
23 |
|
@@ -87,7 +88,7 @@ class ChatBot:
|
|
87 |
self.human_identifier = human_identifier
|
88 |
self.bot_identifier = bot_identifier
|
89 |
|
90 |
-
def chat(self, input_text):
|
91 |
"""
|
92 |
Generates a response from the prompt (and optionally the context) where
|
93 |
it adds the `input_text` as if it was part of the HUMAN dialog
|
|
|
17 |
import argparse
|
18 |
import torch
|
19 |
|
20 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer, PreTrainedModel,\
|
21 |
+
PreTrainedTokenizerBase
|
22 |
from typing import Optional, Union
|
23 |
|
24 |
|
|
|
88 |
self.human_identifier = human_identifier
|
89 |
self.bot_identifier = bot_identifier
|
90 |
|
91 |
+
def chat(self, input_text: str) -> str:
|
92 |
"""
|
93 |
Generates a response from the prompt (and optionally the context) where
|
94 |
it adds the `input_text` as if it was part of the HUMAN dialog
|