Update app.py

app.py

@@ -6,12 +6,11 @@ import torch
 from diffusers import DiffusionPipeline
 from transformers import pipeline
 
-# 번역 파이프라인 초기화
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
-
-dtype = torch.bfloat16
+# 번역 파이프라인 및 하드웨어 설정
 device = "cuda" if torch.cuda.is_available() else "cpu"
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device=device)
 
+dtype = torch.bfloat16
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -40,7 +39,10 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
     ).images[0]
     return image, seed
 
-examples = "Create a new logo for a Tech Startup Design Rocket"
+examples = [
+    ["Create a new logo for a Tech Startup"],
+    ["Design Rocket"]
+]
 
 css = """
 footer {