Update app.py

app.py

@@ -70,6 +70,8 @@ def _getattr(self, name):
     return _orig_getattr(self, name)
 _PrC.__getattribute__ = _getattr
 
+model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
+
 # Florence-2 로드
 device = "cuda" if torch.cuda.is_available() else "cpu"
 scheduler = EulerDiscreteScheduler.from_pretrained(
@@ -84,7 +86,7 @@ tokenizer         = CLIPTokenizer.from_pretrained(
     model_id, subfolder="tokenizer_1"
 )
 feature_extractor = CLIPFeatureExtractor.from_pretrained(
-    model_id, subfolder="feature_extractor"
+    model_repo, subfolder="feature_extractor"
 )
 unet = UNet2DConditionModel.from_pretrained(
     model_id, subfolder="unet", torch_dtype=torch.float16
@@ -98,8 +100,6 @@ florence_processor = AutoProcessor.from_pretrained(LOCAL_FLORENCE, trust_remote_
 
 diffusers.StableDiffusion3Pipeline = StableDiffusionPipeline
 
-model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
-
 pipe = StableDiffusionPipeline(
     text_encoder=text_encoder,
     tokenizer=tokenizer,