update

app.py

@@ -32,7 +32,7 @@ def process_image(image):
     else:
         img = image.convert('RGB')
 
-    predictor = load_predictor('captcha-2000.safetensors')
+    predictor = load_predictor('captcha-7400.safetensors')
     text = predictor.pred_img(img, show=False)
     return text
 

cap.py

@@ -10,11 +10,11 @@ from PIL import Image
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 char_dict = '_0123456789abcdefghijklmnopqrstuvwxyz'
-id_chr_map = {i: c for i, c in enumerate(char_dict)}
+char_dict_pp = '_0123456789abcdefghijklmnopqrstuvwxyz()+-*/='
 
 
 class Predictor:
-    def __init__(self, model_path, char_dict=char_dict):
+    def __init__(self, model_path, char_dict=char_dict_pp):
         self.model = ResnetEncoderDecoder(char_dict).to(device)
         self.model.eval()
         if str(device)=='cpu':