seawolf2357 commited on
Commit
54bb316
·
verified ·
1 Parent(s): 18011ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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", tool="output")
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],