Update app.py

app.py

@@ -165,7 +165,8 @@ def get_image_tensors_batch_cpu(images, image_processor, model=None):
         image_tensors.append(image_tensor)
 
     if len(image_tensors) == 1:
-        return image_tensors[0].unsqueeze(0)
+        img = image_tensors[0].unsqueeze(0)
+        return torch.cat([img, img.clone()], dim=0)
     else:
         return torch.stack(image_tensors, dim=0)
 
@@ -204,15 +205,15 @@ ccd_eval = ccd_eval_cpu_wrapper
 # Global Configuration
 # =========================================
 MODEL_CATALOGUE = {
-    "MAIRA-2 (⚡Recommended for CPU)": "X-iZhang/libra-maira-2",
-    "Libra-v1.0-3B": "X-iZhang/libra-v1.0-3b",
+    "MAIRA-2": "X-iZhang/libra-maira-2",
+    "Libra-v1.0-3B (⚡Recommended for CPU)": "X-iZhang/libra-v1.0-3b",
     "Libra-v1.0-7B": "X-iZhang/libra-v1.0-7b",
     "LLaVA-Med-v1.5": "X-iZhang/libra-llava-med-v1.5-mistral-7b",
     "LLaVA-Rad": "X-iZhang/libra-llava-rad",
     "Med-CXRGen-F": "X-iZhang/Med-CXRGen-F",
     "Med-CXRGen-I": "X-iZhang/Med-CXRGen-I"
 }
-DEFAULT_MODEL_NAME = "MAIRA-2 (⚡Recommended for CPU)"
+DEFAULT_MODEL_NAME = "Libra-v1.0-3B (⚡Recommended for CPU)"
 _loaded_models = {}