diff --git a/QMLS Code/MLMG to MLMV.py b/QMLS Code/MLMG to MLMV.py
index c6cc266..212c062 100644
--- a/QMLS Code/MLMG to MLMV.py	
+++ b/QMLS Code/MLMG to MLMV.py	
@@ -1,5 +1,5 @@
 import torch
-import torch.nn as nn
+import torch.nn as nn 
 import torch.optim as optim
 import numpy as np
 import esm # pre-trained ESM2 models