update direct-load snippet for TabFM_HF

README.md

@@ -54,10 +54,10 @@ preds = reg.predict(X_test)
 You can also load directly using the HuggingFace Hub API:
 
 ```python
-from tabfm.src.pytorch.model import TabFM
+from tabfm.src.pytorch.tabfm_v1_0_0 import TabFM_HF
 
-clf_model = TabFM.from_pretrained("google/tabfm-1.0.0-pytorch", subfolder="classification")
-reg_model = TabFM.from_pretrained("google/tabfm-1.0.0-pytorch", subfolder="regression")
+clf_model = TabFM_HF.from_pretrained("google/tabfm-1.0.0-pytorch", subfolder="classification")
+reg_model = TabFM_HF.from_pretrained("google/tabfm-1.0.0-pytorch", subfolder="regression")
 ```
 
 ### Available Checkpoints