Update README.md

README.md

@@ -31,4 +31,9 @@ outperform all off-the-shelf vision backbones in every downstream setting. In th
 - gains are strongest under frozen-backbone evaluation
 - fine-tuning gains are more dataset-dependent
 - performance remains sensitive to view generation and dataset heterogeneity
-- broader validation on additional surveys and downstream tasks is still needed
+- broader validation on additional surveys and downstream tasks is still needed
+
+## Class Files
+
+HF-style classes for using STRADAVit can be found here:
+https://github.com/andreademarco86/stradavit