Update app.py

app.py

@@ -1,3 +1,13 @@
+from huggingface_hub import snapshot_download
+
+REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
+
+LOCAL_FLORENCE_DIR = snapshot_download(
+    repo_id="microsoft/Florence-2-base",
+    revision=REVISION,
+    local_files_only=False
+)
+
 import sys, types, importlib.machinery, importlib
 
 spec = importlib.machinery.ModuleSpec('flash_attn', loader=None)
@@ -44,8 +54,6 @@ def _getattr(self, name):
     return _orig_getattr(self, name)
 _PrC.__getattribute__ = _getattr
 
-REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
-
 # Florence-2 로드
 device = "cuda" if torch.cuda.is_available() else "cpu"
 florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True, torch_dtype=torch.float16)