chore: Update TashkeelModelEO and TashkeelModelED loading in app.py

app.py

@@ -31,8 +31,10 @@ print('Creating Model...')
 eo_model = TashkeelModelEO(tokenizer, max_seq_len=max_seq_len, n_layers=6, learnable_pos_emb=False)
 ed_model = TashkeelModelED(tokenizer, max_seq_len=max_seq_len, n_layers=3, learnable_pos_emb=False)
 
-eo_model.load_state_dict(torch.load(eo_ckpt_path, map_location=device)).eval().to(device)
-ed_model.load_state_dict(torch.load(ed_ckpt_path, map_location=device)).eval().to(device)
+eo_model.load_state_dict(torch.load(eo_ckpt_path, map_location=device))
+eo_model.eval().to(device)
+ed_model.load_state_dict(torch.load(ed_ckpt_path, map_location=device))
+ed_model.eval().to(device)
 
 @spaces.GPU()
 def infer_catt(input_text, choose_model):