seawolf2357 commited on
Commit
bca41b9
ยท
verified ยท
1 Parent(s): 3689c15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -55,6 +55,11 @@ def generate_images(prompts):
55
  images = [image_pipeline(prompt)[0] for prompt in prompts.split('\n') if prompt.strip() != '']
56
  return images
57
 
 
 
 
 
 
58
  with gr.Blocks() as demo:
59
  gr.Markdown("<h1 align='center'>ํ† ๋ฆฌ์˜ ๋ชจํ—˜: 3D ์• ๋‹ˆ๋ฉ”์ด์…˜ ์ƒ์„ฑ๊ธฐ</h1>")
60
  with gr.Row():
@@ -62,8 +67,9 @@ with gr.Blocks() as demo:
62
  inputs = gr.Textbox(placeholder="์—ฌ๊ธฐ์— ์ž…๋ ฅํ•˜์„ธ์š”.", label="์•„๋™์šฉ ์• ๋‹ˆ๋ฉ”์ด์…˜ ์Šคํฌ๋ฆฝํŠธ๋ฅผ ์ƒ์„ฑํ•˜๊ณ  ์‹ถ์€ ์ฃผ์ œ์–ด๋‚˜ ๋ฌธ์žฅ์„ ์ž…๋ ฅํ•˜์„ธ์š”.")
63
  top_p = gr.Slider(minimum=0, maximum=1.0, value=1.0, step=0.05, label="Top-p (nucleus sampling)")
64
  temperature = gr.Slider(minimum=0, maximum=5.0, value=1.0, step=0.1, label="Temperature")
65
- output = gr.Textbox(label="Generated Script", readonly=True)
66
- prompts_output = gr.TextArea(label="Translated Image Generation Prompts", readonly=True)
 
67
  images_output = gr.Gallery(label="Generated Images")
68
  submit_button = gr.Button("Generate Script")
69
  prompts_button = gr.Button("Translate Prompts")
@@ -73,4 +79,4 @@ with gr.Blocks() as demo:
73
  prompts_button.click(fn=generate_prompts, inputs=[output], outputs=prompts_output)
74
  images_button.click(fn=generate_images, inputs=prompts_output, outputs=images_output)
75
 
76
- demo.launch()
 
55
  images = [image_pipeline(prompt)[0] for prompt in prompts.split('\n') if prompt.strip() != '']
56
  return images
57
 
58
+ # ์Šคํฌ๋ฆฝํŠธ ์ฒ˜๋ฆฌ ๋กœ์ง์„ ํฌํ•จํ•œ generate_prompts ํ•จ์ˆ˜
59
+ def generate_prompts(script):
60
+ # ์Šคํฌ๋ฆฝํŠธ ์ฒ˜๋ฆฌ ๋กœ์ง
61
+ return "์—ฌ๊ธฐ์— ์ฒ˜๋ฆฌ๋œ ํ”„๋กฌํ”„ํŠธ ๋ฐ˜ํ™˜"
62
+
63
  with gr.Blocks() as demo:
64
  gr.Markdown("<h1 align='center'>ํ† ๋ฆฌ์˜ ๋ชจํ—˜: 3D ์• ๋‹ˆ๋ฉ”์ด์…˜ ์ƒ์„ฑ๊ธฐ</h1>")
65
  with gr.Row():
 
67
  inputs = gr.Textbox(placeholder="์—ฌ๊ธฐ์— ์ž…๋ ฅํ•˜์„ธ์š”.", label="์•„๋™์šฉ ์• ๋‹ˆ๋ฉ”์ด์…˜ ์Šคํฌ๋ฆฝํŠธ๋ฅผ ์ƒ์„ฑํ•˜๊ณ  ์‹ถ์€ ์ฃผ์ œ์–ด๋‚˜ ๋ฌธ์žฅ์„ ์ž…๋ ฅํ•˜์„ธ์š”.")
68
  top_p = gr.Slider(minimum=0, maximum=1.0, value=1.0, step=0.05, label="Top-p (nucleus sampling)")
69
  temperature = gr.Slider(minimum=0, maximum=5.0, value=1.0, step=0.1, label="Temperature")
70
+ # Gradio ์ธํ„ฐํŽ˜์ด์Šค ์ดˆ๊ธฐํ™” ๋ถ€๋ถ„์—์„œ readonly ๋งค๊ฐœ๋ณ€์ˆ˜ ์ œ๊ฑฐ
71
+ output = gr.Textbox(label="Generated Script")
72
+ prompts_output = gr.TextArea(label="Translated Image Generation Prompts")
73
  images_output = gr.Gallery(label="Generated Images")
74
  submit_button = gr.Button("Generate Script")
75
  prompts_button = gr.Button("Translate Prompts")
 
79
  prompts_button.click(fn=generate_prompts, inputs=[output], outputs=prompts_output)
80
  images_button.click(fn=generate_images, inputs=prompts_output, outputs=images_output)
81
 
82
+ demo.launch()