correct feedback

app.py

@@ -26,9 +26,8 @@ def chat(message, history=[]):
         last_set_of_ids = history[len(history)-1][2]
         bot_input_ids = torch.cat([last_set_of_ids, new_user_input_ids], dim=-1) 
     else:
-        print("HERE WE GO! ", new_user_input_ids)
         bot_input_ids = new_user_input_ids
-    chat_history_ids = model.generate(bot_input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
+    chat_history_ids = model.generate(bot_input_ids, max_length=5000, pad_token_id=tokenizer.eos_token_id)
     response_ids = chat_history_ids[:, bot_input_ids.shape[-1]:][0]
     response = tokenizer.decode(response_ids, skip_special_tokens=True)
     history.append((message, response, chat_history_ids))
@@ -39,13 +38,12 @@ def feedback(text):
     num_return_sequences=1
     batch =  grammar_tokenizer([text],truncation=True,padding='max_length',max_length=64, return_tensors="pt")
     corrections= grammar_model.generate(**batch,max_length=64,num_beams=2, num_return_sequences=num_return_sequences, temperature=1.5)
+    corrected_text = tokenizer.decode(corrections[0], clean_up_tokenization_spaces=True, skip_special_tokens=True)
     print("The corrections are: ", corrections)
-    if len(corrections) == 0:
+    if corrected_text == text:
         feedback = f'Looks good! Keep up the good work'
     else:
-        suggestion = grammar_tokenizer.batch_decode(corrections[0], skip_special_tokens=True)
-        suggestion = [sug for sug in suggestion if '<' not in sug]
-        feedback = f'\'{" ".join(suggestion)}\' might be a little better'
+        feedback = f'\'{" ".join(corrected_text)}\' might be a little better'
     return feedback
 
 iface = gr.Interface(