Add modeling.py

modeling.py

@@ -1,11 +1,15 @@
-import sys
-from sentence_transformers import SentenceTransformer
-
-# This is required for the Hugging Face model
-def load_model():
-    """Load the ZamAI Multilingual Embeddings model"""
-    return SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')
-
-if __name__ == "__main__":
-    model = load_model()
-    print("ZamAI Multilingual Embeddings model loaded successfully!")
+import sys
+from sentence_transformers import SentenceTransformer
+
+# This is required for the Hugging Face model
+def load_model():
+    """Load the ZamAI Multilingual Embeddings model"""
+    return SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')
+
+if __name__ == "__main__":
+    model = load_model()
+    print("ZamAI Multilingual Embeddings model loaded successfully!")
+    # Save the model weights for Hugging Face upload
+    save_dir = "zamai-weights"
+    model.save_pretrained(save_dir)
+    print(f"Model weights saved to '{save_dir}/'. You can now push this folder to Hugging Face.")