ssourav15 commited on
Commit
84bc648
·
verified ·
1 Parent(s): 655e795

Add model documentation

Browse files
Files changed (1) hide show
  1. README.md +45 -0
README.md ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ title: VaibeSync User Matching Model
3
+ emoji: 💕
4
+ colorFrom: pink
5
+ colorTo: purple
6
+ sdk: pytorch
7
+ tags:
8
+ - dating
9
+ - user-matching
10
+ - recommendation-system
11
+ - pytorch-lightning
12
+ ---
13
+
14
+ # VaibeSync User Matching Model
15
+
16
+ This model powers the VaibeSync dating app's intelligent user matching system.
17
+
18
+ ## Model Details
19
+
20
+ - **Framework**: PyTorch Lightning
21
+ - **Task**: User compatibility prediction
22
+ - **Architecture**: Two-tower neural network
23
+ - **Size**: 0.4 MB
24
+
25
+ ## Usage
26
+
27
+ This model is automatically downloaded and used by the VaibeSync ML API deployed on Railway.
28
+
29
+ ```python
30
+ from models.user_match_two_tower import UserMatchingModel
31
+
32
+ model = UserMatchingModel.load_from_checkpoint("last.ckpt")
33
+ model.eval()
34
+ ```
35
+
36
+ ## Railway Deployment
37
+
38
+ Set these environment variables in your Railway deployment:
39
+
40
+ ```
41
+ MODEL_STORAGE_TYPE=huggingface
42
+ HF_MODEL_REPO=ssourav15/vaibesync-user-matching-model
43
+ ```
44
+
45
+ The model will be automatically downloaded on first startup.