Update app.py

app.py

@@ -141,35 +141,6 @@ run_libra.load_model = safe_load_model
 # (because ccd imports these functions during module load)
 import libra.eval.run_libra as run_libra_module
 
-# Patch get_image_tensors_batch to force CPU
-def get_image_tensors_batch_cpu(images, image_processor, model=None):
-    """CPU-only version of get_image_tensors_batch"""
-    from PIL import Image
-
-    if not isinstance(images, list):
-        images = [images]
-
-    image_tensors = []
-    for image in images:
-        if isinstance(image, str):
-            image = Image.open(image).convert('RGB')
-
-        # Process image
-        if hasattr(image_processor, 'preprocess'):
-            image_tensor = image_processor.preprocess(image, return_tensors='pt')['pixel_values'][0]
-        else:
-            image_tensor = image_processor(image, return_tensors='pt')['pixel_values'][0]
-
-        # Force to CPU (no GPU check)
-        image_tensor = image_tensor.to(device='cpu', dtype=torch.float32)
-        image_tensors.append(image_tensor)
-
-    if len(image_tensors) == 1:
-        img = image_tensors[0].unsqueeze(0)
-        return torch.cat([img, img.clone()], dim=0)
-    else:
-        return torch.stack(image_tensors, dim=0)
-
 # Replace the function in the module BEFORE ccd imports it
 run_libra_module.get_image_tensors_batch = get_image_tensors_batch_cpu
 print('[INFO] Replaced get_image_tensors_batch with CPU-only version')