Update app.py

app.py

@@ -9,13 +9,8 @@ from diffusers import DiffusionPipeline
 from custom_pipeline import FLUXPipelineWithIntermediateOutputs
 from transformers import pipeline
 
-# Hugging Face 토큰 가져오기
-hf_token = os.getenv("HF_TOKEN")
-if not hf_token:
-    raise ValueError("HF_TOKEN environment variable is not set. Please set it to your Hugging Face token.")
-
-# 번역 모델 로드 (토큰 인증 추가)
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", use_auth_token=hf_token)
+# 번역 모델 로드
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
@@ -27,7 +22,7 @@ DEFAULT_INFERENCE_STEPS = 1
 # Device and model setup
 dtype = torch.float16
 pipe = FLUXPipelineWithIntermediateOutputs.from_pretrained(
-    "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, use_auth_token=hf_token
+    "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype
 ).to("cuda")
 torch.cuda.empty_cache()
 
@@ -47,7 +42,7 @@ korean_labels = {
 
 def translate_if_korean(text):
     if any('\u3131' <= char <= '\u3163' or '\uac00' <= char <= '\ud7a3' for char in text):
-        return translator(text, use_auth_token=hf_token)[0]['translation_text']
+        return translator(text)[0]['translation_text']
     return text
 
 # Inference function