Update app.py

app.py

@@ -12,7 +12,7 @@ MODEL.encoder_decoder = transformers.VisionEncoderDecoderModel.from_pretrained(c
 MODEL.tokenizer = MODEL.donut_processor.tokenizer
 
 
-def generate_token_strings(images, skip_special_tokens=True) -> list[str]:
+def generate_token_strings(images, skip_special_tokens=True):
     decoder_output = MODEL.encoder_decoder.generate(
         images,
         max_length=MODEL.encoder_decoder.config.decoder.max_length,
@@ -23,7 +23,7 @@ def generate_token_strings(images, skip_special_tokens=True) -> list[str]:
         decoder_output.sequences, skip_special_tokens=skip_special_tokens
     )
 
-def predict_string(image) -> str:
+def predict_string(image):
     image = MODEL.donut_processor(
         image, random_padding=False, return_tensors="pt"
     ).pixel_values