Update app.py

app.py

@@ -9,15 +9,8 @@ from math import sqrt
 import gradio as gr
 import numpy as np  
     
-model_info = """
-**模型名称**: Google/vit-base-patch16-224  
-**模型介绍**: 本程序根据huggingface上Google开源模型vit，在猫狗图片数据上进行微调,上传一张图片，将会预测其类别并显示结果。模型官网：https://huggingface.co/google/vit-base-patch16-224  
-**程序作者**: 计科三班 王志建、计科三班 罗楷轩  
-**特别支持**: 计科三班 黄成栋  
-"""
-    
 # 加载图像分类模型
-checkpoint_dir = "zhijian12345/checkpoint-905" # 模型检查点目录
+checkpoint_dir = "./checkpoint/checkpoint-905" # 模型检查点目录
 classifier = pipeline("image-classification", model=checkpoint_dir) # 创建图像分类器模型
 vitclassifier = pipeline("image-classification",model="google/vit-base-patch16-224")