LalitMahale commited on
Commit
a3b53c5
·
1 Parent(s): 2b58e71

modified multimodel file

Browse files
Files changed (1) hide show
  1. src/multimodelsearch.py +1 -0
src/multimodelsearch.py CHANGED
@@ -13,6 +13,7 @@ class MultiModelSearch:
13
  images = [
14
  Document(content= f"{document_directory}/{filename}",content_type="image" )
15
  for filename in os.listdir(document_directory)
 
16
  ]
17
 
18
  self.document_stores.write_documents(images)
 
13
  images = [
14
  Document(content= f"{document_directory}/{filename}",content_type="image" )
15
  for filename in os.listdir(document_directory)
16
+ if filename.lower().endswith(('jpg', 'jpeg', 'png'))
17
  ]
18
 
19
  self.document_stores.write_documents(images)