Update app.py

app.py

@@ -13,16 +13,25 @@ from frontPrompt import main as main_f
 
 model_path = "best.pt" 
 modelY = YOLO(model_path)
+os.environ["TRANSFORMERS_CACHE"] = "./.cache"
+cache_folder = "./.cache"
 
-path = "OpenGVLab/InternVL2_5-4B"
+# Load the Hugging Face model and tokenizer globally (downloaded only once)
 model = AutoModel.from_pretrained(
     path,
+    cache_dir=cache_folder,
     torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
-    # load_in_8bit=True,
     low_cpu_mem_usage=True,
     use_flash_attn=True,
-    trust_remote_code=True).eval().to("cpu")
-tokenizer = AutoTokenizer.from_pretrained(path, trust_remote_code=True, use_fast=False)
+    trust_remote_code=True
+).eval().to("cpu")
+
+tokenizer = AutoTokenizer.from_pretrained(
+    path,
+    cache_dir=cache_folder,
+    trust_remote_code=True,
+    use_fast=False
+)
 
 def preprocessing(image):
     """Apply three enhancement filters, including brightness reduction, and resize."""