gorkemgoknar commited on
Commit
fba1393
1 Parent(s): 0311aa1

Update app.py

Browse files

https://www.linkedin.com/pulse/ai-goes-job-interview-g%C3%B6rkem-g%C3%B6knar/

Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -10,12 +10,19 @@ tokenizer = GPT2Tokenizer.from_pretrained('gorkemgoknar/gpt2chatbotenglish')
10
  tokenizer.model_max_length = 1024
11
 
12
 
 
 
13
  def get_chat_response(name, input_txt = "Hello , what is your name?"):
14
- personality = "My name is " + "Gandalf"
 
 
15
 
 
 
16
  bot_input_ids = tokenizer.encode(personality + tokenizer.eos_token + input_txt + tokenizer.eos_token , return_tensors='pt')
17
 
18
  #optimum response and speed
 
19
  chat_history_ids = model.generate(
20
  bot_input_ids, max_length=50,
21
  pad_token_id=tokenizer.eos_token_id,
 
10
  tokenizer.model_max_length = 1024
11
 
12
 
13
+ #See document for experiment https://www.linkedin.com/pulse/ai-goes-job-interview-g%C3%B6rkem-g%C3%B6knar/
14
+
15
  def get_chat_response(name, input_txt = "Hello , what is your name?"):
16
+ #I trained my dataset with "My name is " as prefix, this will let me dump personalities
17
+ #and also gives character sense if of his/her/its name!
18
+ personality = "My name is " + name
19
 
20
+ ##can respond well to history as well but for this quick demo not implemented
21
+ ##see metayazar.com/chatbot for a min 2 history
22
  bot_input_ids = tokenizer.encode(personality + tokenizer.eos_token + input_txt + tokenizer.eos_token , return_tensors='pt')
23
 
24
  #optimum response and speed
25
+ #50 token max length, temperature = 1.3 makes it creative
26
  chat_history_ids = model.generate(
27
  bot_input_ids, max_length=50,
28
  pad_token_id=tokenizer.eos_token_id,