Update app.py

app.py

@@ -55,11 +55,8 @@ def chat_function(Message, Length_of_the_answer, Who_is_next, Base_to_On_subject
             top_k=50,
             top_p=0.9,
             temperature = float(Base_to_On_subject_temperature),                          # 0 for greedy
-            mask_token_id=tokenizer.mask_token_id,
             eos_token_id=tokenizer.eos_token_id,
-            unk_token_id=tokenizer.unk_token_id,
-            pad_token_id=tokenizer.pad_token_id,
-            device='cpu'
+            pad_token_id=tokenizer.pad_token_id
         )
 
         response = tokenizer.decode(chat_history_ids[:, input_len:][0], skip_special_tokens=True)