Update inference.py

inference.py

@@ -5,7 +5,7 @@ def load_model(weights_path="model.pt", device=None):
     if device is None:
         device = "cuda" if torch.cuda.is_available() else "cpu"
     # Replace 'YourModelClass' with the actual class name from model.py
-    model = YourModelClass()  
+    model = IcosahedralRRF()  
     state = torch.load(weights_path, map_location=device)
     model.load_state_dict(state)
     model.to(device)