Update app.py

app.py

@@ -130,11 +130,18 @@ def image_to_3d(trial_id: str, seed: int, randomize_seed: bool, ss_guidance_stre
 
 @spaces.GPU
 def generate_image_from_text(prompt, height, width, guidance_scale, num_steps):
+    # 기본 프롬프트를 추가
+    base_prompt = "wbgmsst, 3D, white background"
+    
+    # 사용자 프롬프트를 번역 (한국어인 경우)
     translated_prompt = translate_if_korean(prompt)
     
+    # 최종 프롬프트 조합
+    final_prompt = f"{translated_prompt}, {base_prompt}"
+    
     with torch.inference_mode():
         image = flux_pipe(
-            prompt=[translated_prompt],
+            prompt=[final_prompt],
             height=height,
             width=width,
             guidance_scale=guidance_scale,