fix load call to match examples

README.md

@@ -34,10 +34,9 @@ pip install tabfm[pytorch]
 ```python
 from tabfm import TabFMClassifier, tabfm_v1_0_0_pytorch as tabfm_v1_0_0
 
-model = tabfm_v1_0_0.load()  # downloads classification weights automatically
+model = tabfm_v1_0_0.load(model_type="classification")
 clf = TabFMClassifier(model=model)
 clf.fit(X_train, y_train)
-preds = clf.predict(X_test)
 probs = clf.predict_proba(X_test)
 ```