NEXAS commited on
Commit
e4e2d04
·
verified ·
1 Parent(s): 4ed9090

Update initate.py

Browse files
Files changed (1) hide show
  1. initate.py +10 -10
initate.py CHANGED
@@ -1,10 +1,10 @@
1
- from utils.ingest_image import extract_and_store_images
2
- from utils.ingest_text import create_vector_database
3
- from utils.ingest_video import intiate_video
4
-
5
- def process_pdf(pdf_path):
6
- image_collection = extract_and_store_images(pdf_path)
7
- text_collection = create_vector_database(pdf_path)
8
- video_collection = intiate_video()
9
- return image_collection, text_collection, video_collection
10
-
 
1
+ from utils.ingest_image import extract_and_store_images
2
+ from utils.ingest_text import create_vector_database
3
+ from utils.ingest_video import intiate_video
4
+
5
+ def process_pdf(pdf_path,temp_folder):
6
+ image_collection = extract_and_store_images(pdf_path)
7
+ text_collection = create_vector_database(pdf_path)
8
+ video_collection = intiate_video(temp_folder)
9
+ return image_collection, text_collection, video_collection
10
+