Update app.py

app.py

@@ -192,8 +192,8 @@ class LW_ViT(nn.Module):
         return x
 
 classifier = LW_ViT()
-classifier.load_state_dict(torch.load("model/chinese_Character_Classification_handwritten_LW_ViT.pth", weights_only=True))
-classifier.to(user_device).eval()
+classifier.load_state_dict(torch.load("model/LW_Vit/chinese_Character_Classification_handwritten_LW_ViT.pth", weights_only=True, map_location=user_device))
+classifier.eval()
 
 """load classes dict"""
 with open("model/chinese_classes.json", "r", encoding="utf-8") as f: