Update app.py
Browse files
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/
|
| 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"""
|