Update app.py

app.py

@@ -17,7 +17,7 @@ model_info = """
 """
     
 # 加载图像分类模型
-checkpoint_dir = "./checkpoint/checkpoint-181" # 模型检查点目录
+checkpoint_dir = "zhijian12345/checkpoint-905" # 模型检查点目录
 classifier = pipeline("image-classification", model=checkpoint_dir) # 创建图像分类器模型
 vitclassifier = pipeline("image-classification",model="google/vit-base-patch16-224")