JJJHHHH commited on
Commit
a7e056b
·
verified ·
1 Parent(s): d563276

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -1
README.md CHANGED
@@ -1,5 +1,12 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
 
3
  ---
4
 
5
  # Model Card: Chinese Calligraphy Character Classifier (ResNet50-based)
@@ -52,4 +59,4 @@ input_tensor = transform(img).unsqueeze(0).to(device)
52
 
53
  outputs = model(input_tensor)
54
  pred_idx = torch.argmax(outputs, dim=1).item()
55
- pred_char = idx2char[pred_idx]
 
1
  ---
2
  license: apache-2.0
3
+ base_model:
4
+ - microsoft/resnet-50
5
+ tags:
6
+ - ocr
7
+ - chinese
8
+ - handwritten-chinese-calligraphy-ocr
9
+ - traditional-chinese-ocr
10
  ---
11
 
12
  # Model Card: Chinese Calligraphy Character Classifier (ResNet50-based)
 
59
 
60
  outputs = model(input_tensor)
61
  pred_idx = torch.argmax(outputs, dim=1).item()
62
+ pred_char = idx2char[pred_idx]