transformers.utils.move_cache()

app.py

@@ -9,9 +9,13 @@ import fitz
 import gradio as gr
 import spaces
 import torch
+import transformers
 from PIL import Image, ImageEnhance
 from transformers import AutoModel, AutoTokenizer
 
+transformers.utils.move_cache()
+transformers.logging.set_verbosity_error()
+
 model_name = "ucaslcl/GOT-OCR2_0"
 device = "cuda" if torch.cuda.is_available() else "cpu"