IncreasingLoss commited on
Commit
5e4e14f
·
verified ·
1 Parent(s): 92a93f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -192,7 +192,7 @@ class LW_ViT(nn.Module):
192
  return x
193
 
194
  classifier = LW_ViT()
195
- classifier.load_state_dict(torch.load("model/LW_Vit/chinese_Character_Classification_handwritten_LW_ViT.pth", weights_only=True, map_location=user_device))
196
  classifier.eval()
197
 
198
  """load classes dict"""
 
192
  return x
193
 
194
  classifier = LW_ViT()
195
+ classifier.load_state_dict(torch.load("model/chinese_Character_Classification_handwritten_LW_ViT.pth", weights_only=True, map_location=user_device))
196
  classifier.eval()
197
 
198
  """load classes dict"""