Update app.py

app.py

@@ -7,13 +7,11 @@ REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
 
 LOCAL_FLORENCE = snapshot_download(
     repo_id="microsoft/Florence-2-base",
-    revision=REVISION,
-    trust_remote_code=True,
+    revision=REVISION
 )
 
 LOCAL_TURBOX = snapshot_download(
-    repo_id="tensorart/stable-diffusion-3.5-large-TurboX",
-    trust_remote_code=True,
+    repo_id="tensorart/stable-diffusion-3.5-large-TurboX"
 )
 
 LOCAL_FLORENCE_DIR = snapshot_download(
@@ -70,10 +68,10 @@ _PrC.__getattribute__ = _getattr
 
 # 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)
+florence_model = AutoModelForCausalLM.from_pretrained(LOCAL_FLORENCE, trust_remote_code=True, torch_dtype=torch.float16)
 florence_model.to("cpu")
 florence_model.eval()
-florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True)
+florence_processor = AutoProcessor.from_pretrained(LOCAL_FLORENCE, trust_remote_code=True)
 
 # Stable Diffusion TurboX 로드
 
@@ -81,7 +79,7 @@ diffusers.StableDiffusion3Pipeline = StableDiffusionPipeline
 
 model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
 pipe = DiffusionPipeline.from_pretrained(
-    model_repo,
+    LOCAL_TURBOX,
     trust_remote_code=True,
     torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
 )