Update app.py

app.py

@@ -1,9 +1,10 @@
+# app.py in Hugging Face Space
 import gradio as gr
 from colpali_engine.models import ColQwen2, ColQwen2Processor
 import torch
 from PIL import Image
 import logging
-import numpy as np
+import os
 
 # Setup logging
 logging.basicConfig(level=logging.INFO)
@@ -12,7 +13,7 @@ logger = logging.getLogger("colqwen-api")
 # Initialize model
 logger.info("Loading ColQwen2 model...")
 model = ColQwen2.from_pretrained(
-    "vidore/colqwen2-v1.0",  # Updated to v1.0
+    "vidore/colqwen2-v1.0",
     torch_dtype=torch.bfloat16,
     device_map="auto",
 )
@@ -23,11 +24,9 @@ logger.info("Model loaded successfully")
 def process_image(image):
     try:
         logger.info("Processing image")
-        # Convert to PIL Image if needed
         if not isinstance(image, Image.Image):
             image = Image.fromarray(image)
         
-        # Process image
         inputs = processor(
             images=image,
             return_tensors="pt"
@@ -55,5 +54,5 @@ interface = gr.Interface(
     description="Generate embeddings from images using ColQwen2"
 )
 
-# Launch with API
-interface.launch(server_name="0.0.0.0", server_port=7861)
+# Let Gradio choose an available port
+interface.launch(server_name="0.0.0.0")