gorkemgoknar commited on
Commit
ef5c315
1 Parent(s): 7707a25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -24,20 +24,20 @@ SPECIAL_TOKENS = ["<bos>", "<eos>", "<speaker1>", "<speaker2>", "<pad>"]
24
  #See document for experiment https://www.linkedin.com/pulse/ai-goes-job-interview-g%C3%B6rkem-g%C3%B6knar/
25
 
26
 
27
-
28
- def get_chat_response(name, input_txt = "Hello , what is your name?",history=[]):
29
 
30
- history.append(input_txt)
31
- history_e = [tokenizer.encode(e) for e in history]
 
 
32
 
33
  personality = "My name is " + name
34
 
35
  bos, eos, speaker1, speaker2 = tokenizer.convert_tokens_to_ids(SPECIAL_TOKENS[:-1])
36
- sequence = [[bos] + tokenizer.encode(personality)] + history_e
37
  sequence = [sequence[0]] + [[speaker2 if (len(sequence)-i) % 2 else speaker1] + s for i, s in enumerate(sequence[1:])]
38
  sequence = list(chain(*sequence))
39
 
40
-
41
  #bot_input_ids = tokenizer.encode(personality + tokenizer.eos_token + input_txt + tokenizer.eos_token , return_tensors='pt')
42
  sequence_len = len(sequence)
43
 
 
24
  #See document for experiment https://www.linkedin.com/pulse/ai-goes-job-interview-g%C3%B6rkem-g%C3%B6knar/
25
 
26
 
27
+ def get_chat_response(name,history=[], input_txt = "Hello , what is your name?"):
 
28
 
29
+ ai_history = history.copy()
30
+
31
+ ai_history.append(input_txt)
32
+ ai_history_e = [tokenizer.encode(e) for e in ai_history]
33
 
34
  personality = "My name is " + name
35
 
36
  bos, eos, speaker1, speaker2 = tokenizer.convert_tokens_to_ids(SPECIAL_TOKENS[:-1])
37
+ sequence = [[bos] + tokenizer.encode(personality)] + ai_history_e
38
  sequence = [sequence[0]] + [[speaker2 if (len(sequence)-i) % 2 else speaker1] + s for i, s in enumerate(sequence[1:])]
39
  sequence = list(chain(*sequence))
40
 
 
41
  #bot_input_ids = tokenizer.encode(personality + tokenizer.eos_token + input_txt + tokenizer.eos_token , return_tensors='pt')
42
  sequence_len = len(sequence)
43