Update preprocess_test.py

preprocess_test.py

@@ -36,6 +36,8 @@ class Preprocess_Test:
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         # self.output_path=output_path
         print("INSIDE CLEANING GOT THE DATASET")
+        import __main__
+        __main__.Model = Model
         
     def delete_redundant(self,percent):
         cols_to_be_deleted=[]