progress

app.py

@@ -53,9 +53,9 @@ def respond(input):
         # val, ix = out[:, -1].data.topk(1)
 
         # outputs[i] = ix[0][0]
-        outputs[i] = ix
+        outputs[i] = ix[0]
         # if ix[0][0] == vocab_token_dict['<eos>']:
-        if ix == vocab_token_dict['<eos>']:
+        if ix[0] == vocab_token_dict['<eos>']:
             break
     return ' '.join([indices_to_tokens[ix] for ix in outputs[1:i]])