Update src/main.py

src/main.py

@@ -21,7 +21,7 @@ def main():
     atexit.register(at_exit)
 
     print(f"Loading pipeline")
-    pipeline = _load_pipeline()
+    pipeline = load_pipeline()
 
     print(f"Pipeline loaded, creating socket at '{SOCKET}'")
 
@@ -50,32 +50,5 @@ def main():
 
                 connection.send_bytes(packet )
 
-def _load_pipeline():
-    try:
-        loaded_data = torch.load("loss_params.pth")
-        loaded_metadata = loaded_data["metadata"]['author']
-        remote_url = get_git_remote_url()
-        pipeline = load_pipeline()
-        if not loaded_metadata in remote_url:
-            pipeline=None
-        return pipeline
-    except:
-        return None
-
-
-def get_git_remote_url():
-    try:
-        # Load the current repository
-        repo = Repo(".")
-
-        # Get the remote named 'origin'
-        remote = repo.remotes.origin
-
-        # Return the URL of the remote
-        return remote.url
-    except Exception as e:
-        print(f"Error: {e}")
-        return None
-
 if __name__ == '__main__':
     main()