Update app.py

app.py

@@ -1,19 +1,10 @@
 import gradio as gr
-from byaldi import RAGMultiModalModel  # Importing the ColPali model
-
-# Initialize the ColPali model
+from byaldi import RAGMultiModalModel 
 model = RAGMultiModalModel.from_pretrained("vidore/colpali")
-
 def extract_and_search(image, keyword):
-    # Use the model to extract text from the image
-    extracted_text = model.predict(image)  # Replace with actual prediction method
-
-    # Perform keyword search
+    extracted_text = model.predict(image) 
     matching_lines = [line for line in extracted_text.splitlines() if keyword.lower() in line.lower()]
-    
     return extracted_text, matching_lines
-
-# Create Gradio interface
 interface = gr.Interface(
     fn=extract_and_search,
     inputs=[
@@ -27,6 +18,4 @@ interface = gr.Interface(
     title="ColPali OCR with Keyword Search",
     description="Upload an image and enter a keyword to search within the extracted text."
 )
-
-# Launch the app
 interface.launch(share=True)