fix inference_app.py

inference_app.py

@@ -104,7 +104,6 @@ def get_system(system_id: str) -> PinderSystem:
     return PinderSystem(system_id)
 from Bio import PDB
 from Bio.PDB.PDBIO import PDBIO
-from torch.distributed.fsdp import FullyShardedDataParallel as FSDP
 
 def extract_coordinates_from_pdb(filename):
     r"""
@@ -874,14 +873,8 @@ class FinalMPNNModelight(pl.LightningModule):
             },
         }
 
-# model_path = "/home/sukanya/iitm_bisect_pinder_submission/EquiMPNN-epoch=413-val_loss=9.25-val_acc=0.00.ckpt"
-# model = FinalMPNNModelight.load_from_checkpoint(model_path)
-model = FinalMPNNModelight()
-model = FSDP(model)
-checkpoint = torch.load("/home/sukanya/iitm_bisect_pinder_submission/EquiMPNN-epoch=413-val_loss=9.25-val_acc=0.00.ckpt")
-model_state_dict = checkpoint['state_dict'] 
-model.load_state_dict(model_state_dict)
-
+model_path = "/home/sukanya/iitm_bisect_pinder_submission/EquiMPNN-epoch=413-val_loss=9.25-val_acc=0.00.ckpt"
+model = FinalMPNNModelight.load_from_checkpoint(model_path)
 trainer = pl.Trainer(