update of the model to take images and text

e906a7e0 · jrzaurin · bcffd7e2 · e906a7e0
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

prepare_data.py prepare_data.py +1 -1

未找到文件。
--- a/prepare_data.py
+++ b/prepare_data.py
@@ -363,7 +363,7 @@ if __name__ == '__main__':

        wd_dataset_airbnb = prepare_data_airbnb(
            # df = DF.sample(5000),
-            df = DF.sample,
+            df = DF,
            img_id = img_id,
            img_path = img_path,
            text_col = text_col,