run480 commited on
Commit
3627840
1 Parent(s): 415835e

Update app.py

Browse files

Fix syntax errors

Files changed (1) hide show
  1. app.py +25 -24
app.py CHANGED
@@ -406,30 +406,31 @@ mdl = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
406
 
407
  def converse(user_input, chat_history=[]):
408
  user_input_ids = chat_tkn(user_input + chat_tkn.eos_token, return_tensors='pt').input_ids
409
-     # keep history in the tensor
410
-     bot_input_ids = torch.cat([torch.LongTensor(chat_history), user_input_ids], dim=-1)
411
-     # get response
412
-     chat_history = mdl.generate(bot_input_ids, max_length=1000, pad_token_id=chat_tkn.eos_token_id).tolist()
413
-     print (chat_history)
414
-     response = chat_tkn.decode(chat_history[0]).split("<|endoftext|>")
415
-     print("starting to print response")
416
-     print(response)
417
-     # html for display
418
-     html = "<div class='mybot'>"
419
-     for x, mesg in enumerate(response):
420
-         if x%2!=0 :
421
-            mesg="Alicia:"+mesg
422
-            clazz="alicia"
423
-         else :
424
-            clazz="user"
425
-         print("value of x")
426
-         print(x)
427
-         print("message")
428
-         print (mesg)
429
-         html += "<div class='mesg {}'> {}</div>".format(clazz, mesg)
430
-     html += "</div>"
431
-     print(html)
432
-     return html, chat_history
 
433
 
434
  import gradio as grad
435
 
 
406
 
407
  def converse(user_input, chat_history=[]):
408
  user_input_ids = chat_tkn(user_input + chat_tkn.eos_token, return_tensors='pt').input_ids
409
+ # keep history in the tensor
410
+ bot_input_ids = torch.cat([torch.LongTensor(chat_history), user_input_ids], dim=-1)
411
+ # get response
412
+ chat_history = mdl.generate(bot_input_ids, max_length=1000, pad_token_id=chat_tkn.eos_token_id).tolist()
413
+ print (chat_history)
414
+ response = chat_tkn.decode(chat_history[0]).split("<|endoftext|>")
415
+ print("starting to print response")
416
+ print(response)
417
+ # html for display
418
+ html = "<div class='mybot'>"
419
+ for x, mesg in enumerate(response):
420
+ if x%2!=0 :
421
+ mesg="Alicia:"+mesg
422
+ clazz="alicia"
423
+ else :
424
+ clazz="user"
425
+ print("value of x")
426
+ print(x)
427
+ print("message")
428
+ print (mesg)
429
+ html += "<div class='mesg {}'> {}</div>".format(clazz, mesg)
430
+ html += "</div>"
431
+ print(html)
432
+
433
+ return html, chat_history
434
 
435
  import gradio as grad
436