diff --git a/misinformation/test/test_multimodal_search.py b/misinformation/test/test_multimodal_search.py index f5310a8..6ed4a41 100644 --- a/misinformation/test/test_multimodal_search.py +++ b/misinformation/test/test_multimodal_search.py @@ -22,27 +22,9 @@ def test_read_img(): test_img = ms.read_img(TEST_IMAGE_2) assert numpy.array(test_img)[257][34] == [70, 66, 63] - - -#def test_load_feature_extractor_model_blip2(): -# multimodal_device = device("cuda" if cuda.is_available() else "cpu") -# (model, vis_processors, txt_processors,) = ms.load_feature_extractor_model_blip2(multimodal_device) - - - - - - - - - - - - - - - - + # def test_load_feature_extractor_model_blip2(): + # multimodal_device = device("cuda" if cuda.is_available() else "cpu") + # (model, vis_processors, txt_processors,) = ms.load_feature_extractor_model_blip2(multimodal_device) mydict = {} for img_path in images: @@ -54,4 +36,4 @@ def test_read_img(): keys = list(mydict.keys()) assert len(mydict) == 12 for key in keys: - assert len(mydict[key]["3_non-deterministic summary"]) == 3 \ No newline at end of file + assert len(mydict[key]["3_non-deterministic summary"]) == 3