fix: model caching bug

Signed-off-by: Ubuntu <[email protected]>

app.py

@@ -24,20 +24,19 @@ def load_model(pretrained_revision: str = 'main'):
         f"Loading model checkpoint from repo: {pretrained_repo_name}, revision: {pretrained_revision}")
     if processor is None or loaded_revision is None or loaded_revision != pretrained_revision:
         loaded_revision = pretrained_revision
-
-    processor = DonutProcessor.from_pretrained(
-        pretrained_repo_name, revision=pretrained_revision)  # , use_auth_token="...")
-    processor.image_processor.do_align_long_axis = False
-    # do not manipulate image size and position
-    processor.image_processor.do_resize = False
-    processor.image_processor.do_thumbnail = False
-    processor.image_processor.do_pad = False
-    # processor.image_processor.do_rescale = False
-    processor.image_processor.do_normalize = True
-    print(f'processor image size: {processor.image_processor.size}')
-    model = VisionEncoderDecoderModel.from_pretrained(
-        pretrained_repo_name, revision=pretrained_revision)  # use_auth_token="...",
-    print(f'model checkpoint loaded')
+        processor = DonutProcessor.from_pretrained(
+            pretrained_repo_name, revision=pretrained_revision)  # , use_auth_token="...")
+        processor.image_processor.do_align_long_axis = False
+        # do not manipulate image size and position
+        processor.image_processor.do_resize = False
+        processor.image_processor.do_thumbnail = False
+        processor.image_processor.do_pad = False
+        # processor.image_processor.do_rescale = False
+        processor.image_processor.do_normalize = True
+        print(f'processor image size: {processor.image_processor.size}')
+        model = VisionEncoderDecoderModel.from_pretrained(
+            pretrained_repo_name, revision=pretrained_revision)  # use_auth_token="...",
+        print(f'model checkpoint loaded')
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model.to(device)