Update app.py

app.py

@@ -28,13 +28,17 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 model = model.to(device)
 
 def load_image(image_file):
-    """加载本地图片或 URL 图片"""
-    if isinstance(image_file, str) and (image_file.startswith('http') or image_file.startswith('https')):
+    """确保 image 是 `PIL.Image`"""
+    if isinstance(image_file, Image.Image):
+        return image_file.convert("RGB")  # 直接返回 `PIL.Image`
+
+    elif isinstance(image_file, str) and (image_file.startswith('http') or image_file.startswith('https')):
         response = requests.get(image_file)
-        image = Image.open(BytesIO(response.content)).convert('RGB')
-    else:
-        image = Image.open(image_file).convert('RGB')
-    return image
+        return Image.open(BytesIO(response.content)).convert('RGB')
+
+    else:  # 这里如果 `image_file` 是路径
+        return Image.open(image_file).convert("RGB")
+
 
 def llava_infer(image, text, temperature, top_p, max_tokens):
     """LLaVA 模型推理"""