Kazel commited on
Commit
1479553
·
1 Parent(s): c575567
Files changed (2) hide show
  1. app.py +2 -2
  2. rag.py +1 -0
app.py CHANGED
@@ -233,7 +233,7 @@ def create_ui():
233
  gr.Markdown("# Collar Multimodal RAG Demo")
234
  gr.Markdown("Settings Available On Local Offline Setup")
235
 
236
- with gr.Tab("Upload PDF"):
237
  with gr.Column():
238
  max_pages_input = gr.Slider(
239
  minimum=1,
@@ -242,7 +242,7 @@ def create_ui():
242
  step=10,
243
  label="Max pages to extract and index per document"
244
  )
245
- file_input = gr.Files(label="Upload PDFs")
246
  file_list = gr.Textbox(label="Uploaded Files", interactive=False, value="Available on Local Setup")
247
  status = gr.Textbox(label="Indexing Status", interactive=False)
248
 
 
233
  gr.Markdown("# Collar Multimodal RAG Demo")
234
  gr.Markdown("Settings Available On Local Offline Setup")
235
 
236
+ with gr.Tab("Upload Documents"):
237
  with gr.Column():
238
  max_pages_input = gr.Slider(
239
  minimum=1,
 
242
  step=10,
243
  label="Max pages to extract and index per document"
244
  )
245
+ file_input = gr.Files(label="Upload PPTs/PDFs")
246
  file_list = gr.Textbox(label="Uploaded Files", interactive=False, value="Available on Local Setup")
247
  status = gr.Textbox(label="Indexing Status", interactive=False)
248
 
rag.py CHANGED
@@ -86,6 +86,7 @@ class Rag:
86
 
87
  # Close model thread (colpali)
88
  print(f"Querying OpenAI for query={query}, imagesPaths={imagesPaths}")
 
89
 
90
  try:
91
 
 
86
 
87
  # Close model thread (colpali)
88
  print(f"Querying OpenAI for query={query}, imagesPaths={imagesPaths}")
89
+ from ollama import chat
90
 
91
  try:
92