Update app.py

app.py

@@ -63,7 +63,8 @@ def translate_to_english(text: str) -> str:
     except Exception as e:
         print(f"Translation error: {str(e)}")
         return text
-        
+
+
 # FLUX 파이프라인 초기화 부분 수정
 print("Initializing FLUX pipeline...")
 try:
@@ -90,23 +91,21 @@ except Exception as e:
     print(f"Error initializing FLUX pipeline: {str(e)}")
     raise
 
-# LoRA 가중치 로드 부분
+# LoRA 가중치 로드 부분 수정
 print("Loading LoRA weights...")
 try:
-    lora_path = hf_hub_download(
-        repo_id="openfree/myt-flux-fantasy",
-        filename="myt-flux-fantasy.safetensors",
-        use_auth_token=HF_TOKEN
-    )
-    print(f"LoRA weights downloaded to: {lora_path}")
+    # 로컬 LoRA 파일 경로 지정
+    lora_path = "myt-flux-fantasy.safetensors"  # 현재 디렉토리의 파일 직접 사용
     
+    print(f"Loading LoRA weights from local file: {lora_path}")
     pipe.load_lora_weights(lora_path)
     pipe.fuse_lora(lora_scale=0.125)
     print("LoRA weights loaded and fused successfully")
 
 except Exception as e:
     print(f"Error loading LoRA weights: {str(e)}")
-    raise ValueError("Failed to load LoRA weights")
+    raise ValueError(f"Failed to load LoRA weights from {lora_path}: {str(e)}")
+    
 
 # generate_image 함수 수정
 @spaces.GPU(duration=60)
@@ -299,7 +298,7 @@ footer {display: none}
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
         <div class="main-title">
-            <h1>🎨 Webtoon Canvas</h1>
+            <h1>🎨 Webtoon Studio</h1>
             <p>Generate webtoon-style images and add text with various styles and positions.</p>
         </div>
     """)