ALEXFLR commited on
Commit
457482e
·
verified ·
1 Parent(s): 288f3fd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -8,7 +8,8 @@ tags:
8
  - bilinear-networks
9
  metrics:
10
  - loss
11
- pipeline_tag: table-regression
 
12
  ---
13
 
14
  # SBL-NET (Scalar Bilinear Linear Network)
@@ -49,7 +50,7 @@ import torch as t
49
  from huggingface_hub import hf_hub_download
50
 
51
  # 1. Download the architecture and weights files (replace YOUR_USERNAME)
52
- REPO_ID = "YOUR_USERNAME/sbl-net"
53
  hf_hub_download(repo_id=REPO_ID, filename="model.py", local_dir=".")
54
  hf_hub_download(repo_id=REPO_ID, filename="model_weights_hybrid.pth", local_dir=".")
55
 
 
8
  - bilinear-networks
9
  metrics:
10
  - loss
11
+ pipeline_tag: tabular-regression
12
+
13
  ---
14
 
15
  # SBL-NET (Scalar Bilinear Linear Network)
 
50
  from huggingface_hub import hf_hub_download
51
 
52
  # 1. Download the architecture and weights files (replace YOUR_USERNAME)
53
+ REPO_ID = "ALEX_FLR/sbl-net"
54
  hf_hub_download(repo_id=REPO_ID, filename="model.py", local_dir=".")
55
  hf_hub_download(repo_id=REPO_ID, filename="model_weights_hybrid.pth", local_dir=".")
56