Update inference_img.py

inference_img.py

@@ -5,9 +5,7 @@ import argparse
 from torch.nn import functional as F
 import warnings
 
-
-        
-    warnings.filterwarnings("ignore")
+warnings.filterwarnings("ignore")
     
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     torch.set_grad_enabled(False)
@@ -118,4 +116,3 @@ import warnings
             print(f"Saving to {save_path} → success: {success}")
             
     print("Checkpoint reached!")
-