Update app.py

app.py

@@ -63,8 +63,13 @@ def process_image(model, tokenizer, transform, device, check_type, image, text):
         target_ratio = (1, 1)
 
     # 文本处理
-    text += ' '
-    input_ids = tokenizer(text)['input_ids'][1:]
+    text_input = text
+
+    if text_input[0] in '!"#$%&\'()*+,-./0123456789:;<=>?@^_{|}~0123456789':
+        input_ids = tokenizer(text_input)['input_ids'][1:]
+    else:
+        input_ids = tokenizer(' '+text_input)['input_ids'][1:]
+
     input_ids = torch.tensor(input_ids, device=device)
     
     # 获取嵌入