fix + 2

app.py

@@ -81,6 +81,10 @@ transform = transforms.Compose([
 # 定义预测函数
 def classify_image(image):
     # image 已经是一个 PIL 图像
+    # 将图像转换为灰度模式
+    if not isinstance(image, Image.Image):
+        raise TypeError(f"Expected image to be PIL Image, but got {type(image)}")
+    
     # 将图像转换为灰度模式
     image = image.convert("L")
     
@@ -117,10 +121,10 @@ def classify_image(image):
 
 iface = gr.Interface(
     fn=classify_image,
-    inputs=gr.Sketchpad(crop_size=(256,256), type='numpy', image_mode='L', brush=gr.Brush()),
+    inputs=gr.Sketchpad(crop_size=(256,256), type='pil', image_mode='L', brush=gr.Brush()),
     outputs=gr.Label(num_top_classes=1),
     title="MNIST Digit Classification with ViT", 
-    description="使用鼠标手绘一个数字，模型将预测其所属的类别。"
+    description="Use the mouse to hand draw a number and the model will predict the category it belongs to."
 )