Update app/main.py

app/main.py

@@ -66,6 +66,11 @@ with gr.Blocks(title="HTRflow", theme=theme, css=css) as demo:
         if os.path.exists(tmp_dir) and os.path.isdir(tmp_dir):
             shutil.rmtree(tmp_dir)
 
+
+    @demo.load()
+    def inital_trocr_load():
+        TrOCR("Riksarkivet/trocr-base-handwritten-hist-swe-2")
+    
     @demo.load(
         inputs=[template_output_yaml_code],
         outputs=[template_output_yaml_code],
@@ -73,9 +78,6 @@ with gr.Blocks(title="HTRflow", theme=theme, css=css) as demo:
     def inital_yaml_code(template_output_yaml_code):
         return template_output_yaml_code
 
-    @demo.load()
-    def inital_trocr_load():
-        return TrOCR("Riksarkivet/trocr-base-handwritten-hist-swe-2")
 
     def sync_gradio_objects(input_value, state_value):
         """Synchronize the YAML state if there is a mismatch."""