Spaces:
Sleeping
Sleeping
seawolf2357
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ model.eval()
|
|
57 |
model = model.cuda()
|
58 |
|
59 |
# 번역 모델 초기화
|
60 |
-
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
|
61 |
|
62 |
# FLUX 파이프라인 초기화 부분 수정
|
63 |
flux_pipe = FluxPipeline.from_pretrained(
|
@@ -168,7 +168,7 @@ with gr.Blocks(analytics_enabled=False, css=css) as dynamicrafter_iface:
|
|
168 |
img_seed = gr.Slider(label='Random Seed', minimum=0, maximum=10000, step=1, value=123)
|
169 |
img_generate_btn = gr.Button("Generate Image")
|
170 |
with gr.Row():
|
171 |
-
img_output_image = gr.Image(label="Generated Image"
|
172 |
|
173 |
img_generate_btn.click(
|
174 |
inputs=[img_input_text, img_seed],
|
|
|
57 |
model = model.cuda()
|
58 |
|
59 |
# 번역 모델 초기화
|
60 |
+
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device=0) # GPU 사용 설정
|
61 |
|
62 |
# FLUX 파이프라인 초기화 부분 수정
|
63 |
flux_pipe = FluxPipeline.from_pretrained(
|
|
|
168 |
img_seed = gr.Slider(label='Random Seed', minimum=0, maximum=10000, step=1, value=123)
|
169 |
img_generate_btn = gr.Button("Generate Image")
|
170 |
with gr.Row():
|
171 |
+
img_output_image = gr.Image(label="Generated Image")
|
172 |
|
173 |
img_generate_btn.click(
|
174 |
inputs=[img_input_text, img_seed],
|