fix + 3

app.py

@@ -80,8 +80,11 @@ transform = transforms.Compose([
 
 # 定义预测函数
 def classify_image(image):
-    # image 已经是一个 PIL 图像
-    # 将图像转换为灰度模式
+    # 检查是否包含 'composite' 数据
+    if isinstance(image, dict) and 'composite' in image:
+        image = image['composite']
+    
+    # 确保 image 是一个 PIL 图像
     if not isinstance(image, Image.Image):
         raise TypeError(f"Expected image to be PIL Image, but got {type(image)}")
     
@@ -109,7 +112,6 @@ def classify_image(image):
     # 返回结果字典，包含预测类别和置信度
     return {str(predicted.item()): confidence}
 
-
 # # 创建Gradio界面
 # iface = gr.Interface(
 #     fn=classify_image,