Spaces:
Running
Running
gorkemgoknar
commited on
Commit
•
fba1393
1
Parent(s):
0311aa1
Update app.py
Browse fileshttps://www.linkedin.com/pulse/ai-goes-job-interview-g%C3%B6rkem-g%C3%B6knar/
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 |
-
|
|
|
|
|
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,
|