Upload model.py with huggingface_hub

model.py

@@ -109,6 +109,27 @@ class MultiModalEnsembleC(nn.Module):
         with open(config_path, "w") as f:
             json.dump(config, f)
 
+    @classmethod
+    def from_pretrained(cls, load_directory):
+        """
+        Load model weights and configuration from a directory.
+        
+        Args:
+            load_directory (str): Path to the directory containing model weights and configuration.
+        
+        Returns:
+            MultiModalEnsembleC: A model instance with loaded weights.
+        """
+        config_path = os.path.join(load_directory, "config.json")
+        with open(config_path, "r") as f:
+            config = json.load(f)
+
+        # Create the model instance
+        model = cls(num_classes=config["num_classes"])
+        model_path = os.path.join(load_directory, "pytorch_model.bin")
+        model.load_state_dict(torch.load(model_path))
+        return model
+    
     def forward(self, x, y, z):
         """
         Forward pass of the MultiModalEnsembleC model.