Hugging Face's logo

Spaces:
zhijian12345
/
cat_classifiter
Sleeping

App Files Community
zhijian12345 commited on
Commit
3bdb19a
•
1 Parent(s): 1d56b50

Update checkpoint

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import gradio as gr
10
  import numpy as np
11
 
12
  # 加载图像分类模型
13
- checkpoint_dir = "./checkpoint/checkpoint-905" # 模型检查点目录
14
  classifier = pipeline("image-classification", model=checkpoint_dir) # 创建图像分类器模型
15
  vitclassifier = pipeline("image-classification",model="google/vit-base-patch16-224")
16
 
 
10
  import numpy as np
11
 
12
  # 加载图像分类模型
13
+ checkpoint_dir = "./checkpoint-905" # 模型检查点目录
14
  classifier = pipeline("image-classification", model=checkpoint_dir) # 创建图像分类器模型
15
  vitclassifier = pipeline("image-classification",model="google/vit-base-patch16-224")
16