Update app.py

app.py

@@ -20,12 +20,12 @@ def install_fa2():
 
 
 model = ColQwen2.from_pretrained(
-        "manu/colqwen2-v1.0-alpha",
+        "vidore/colqwen2-v1.0",
         torch_dtype=torch.bfloat16,
         device_map="cuda:0",  # or "mps" if on Apple Silicon
         # attn_implementation="flash_attention_2", # should work on A100
     ).eval()
-processor = ColQwen2Processor.from_pretrained("manu/colqwen2-v1.0")
+processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v1.0")
 
 
 def encode_image_to_base64(image):
@@ -39,7 +39,7 @@ def query_gpt4o_mini(query, images):
     """Calls OpenAI's GPT-4o-mini with the query and image data."""
     from openai import OpenAI
 
-    images = [encode_image_to_base64(image) for image in images]
+    base64_images = [encode_image_to_base64(image) for image in images]
     client = OpenAI(api_key=os.env.get("OPENAI_KEY"))
     PROMPT = """
     You are a smart assistant designed to answer questions about a PDF document.
@@ -122,7 +122,7 @@ def search(query: str, ds, images, k):
         results.append((images[idx], f"Page {idx}"))
 
     # Generate response from GPT-4o-mini
-    ai_response = "Activate AI response by forking and adding your GPT-4o key" # query_gpt4o_mini(query, results)
+    ai_response = query_gpt4o_mini(query, results)
 
     return results, ai_response