crscardellino commited on
Commit
2c5b87c
1 Parent(s): 9591b5b

Fixed some style validations on chatbot.py

Browse files
Files changed (1) hide show
  1. 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, PreTrainedTokenizerBase
 
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