Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
-from byaldi import RAGMultiModalModel  # Importing the ColPali model
+from transformers import RAGMultiModalModel  # Importing the ColPali model
 
 # Initialize the ColPali model
 model = RAGMultiModalModel.from_pretrained("vidore/colpali-v1.2")
 
 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
+    inputs = {"images": [image]}
+    extracted_text = model.generate(**inputs)  # Replace with actual prediction method
 
     # Perform keyword search
     matching_lines = [line for line in extracted_text.splitlines() if keyword.lower() in line.lower()]
@@ -29,4 +30,4 @@ interface = gr.Interface(
 )
 
 # Launch the app
-interface.launch(share=True)
+interface.launch(share=True)