seawolf2357 commited on
Commit
faf98f2
ยท
verified ยท
1 Parent(s): d9f0669

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +37 -1
app.py CHANGED
@@ -1,3 +1,39 @@
1
  import gradio as gr
2
 
3
- gr.Interface.load("models/goofyai/3d_render_style_xl").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
 
3
+ # ์ด ํ•จ์ˆ˜๋Š” ์‚ฌ์šฉ์ž ์ž…๋ ฅ์— ๋Œ€ํ•œ ๋‚˜๋ ˆ์ด์…˜ ์Šคํฌ๋ฆฝํŠธ๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
4
+ # ์‹ค์ œ ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜์—์„œ๋Š” OpenAI API๋‚˜ ๋‹ค๋ฅธ ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•ด ๊ตฌํ˜„ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
5
+ def generate_script(inputs):
6
+ # ์˜ˆ์‹œ๋ฅผ ์œ„ํ•œ ๊ฐ€์ƒ์˜ ์Šคํฌ๋ฆฝํŠธ ๋ฐ˜ํ™˜
7
+ return "์ด๊ณณ์€ ํ™”๋ คํ•œ ๋„์‹œ์˜ ์ „๊ฒฝ์ž…๋‹ˆ๋‹ค.\n์ € ๋ฉ€๋ฆฌ ๋น›๋‚˜๋Š” ๋นŒ๋”ฉ๋“ค ์‚ฌ์ด๋กœ ํ† ๋ฆฌ๊ฐ€ ๋‚ ์•„๊ฐ‘๋‹ˆ๋‹ค."
8
+
9
+ # Gradio ์ธํ„ฐํŽ˜์ด์Šค๋ฅผ ๋กœ๋“œํ•˜๋Š” ํ•จ์ˆ˜
10
+ def load_interface():
11
+ return gr.Interface.load("models/goofyai/3d_render_style_xl")
12
+
13
+ # ์Šคํฌ๋ฆฝํŠธ์˜ ๊ฐ ์ค„์„ ์ด๋ฏธ์ง€๋กœ ๋ณ€ํ™˜ํ•˜๋Š” ํ•จ์ˆ˜
14
+ def script_to_images(script):
15
+ # ๋ฏธ๋ฆฌ ๋กœ๋“œํ•œ Gradio ์ธํ„ฐํŽ˜์ด์Šค๋ฅผ ์‚ฌ์šฉ
16
+ iface = load_interface()
17
+ images = []
18
+ for line in script.split('\n'):
19
+ # ์ด๋ฏธ์ง€ ์ƒ์„ฑ ๋ชจ๋ธ์— ๊ฐ ์Šคํฌ๋ฆฝํŠธ ์ค„์„ ์ž…๋ ฅ์œผ๋กœ ์ œ๊ณต
20
+ image = iface(line)
21
+ images.append(image)
22
+ return images
23
+
24
+ with gr.Blocks() as demo:
25
+ with gr.Row():
26
+ inputs = gr.Textbox(label="๋‚˜๋ ˆ์ด์…˜ ์Šคํฌ๋ฆฝํŠธ ์ž…๋ ฅ")
27
+ generate_btn = gr.Button("๋‚˜๋ ˆ์ด์…˜ ์Šคํฌ๋ฆฝํŠธ ์ƒ์„ฑ")
28
+ with gr.Row():
29
+ script_output = gr.Textbox(label="์ƒ์„ฑ๋œ ์Šคํฌ๋ฆฝํŠธ", interactive=False)
30
+ with gr.Row():
31
+ images_output = gr.Gallery(label="์ƒ์„ฑ๋œ ์ด๋ฏธ์ง€")
32
+
33
+ # ์Šคํฌ๋ฆฝํŠธ ์ƒ์„ฑ ๋ฒ„ํŠผ ํด๋ฆญ ์‹œ ์ด๋ฒคํŠธ ํ•ธ๋“ค๋Ÿฌ
34
+ generate_btn.click(fn=generate_script, inputs=inputs, outputs=script_output)
35
+
36
+ # ์Šคํฌ๋ฆฝํŠธ ์ถœ๋ ฅ๊ฐ’ ๋ณ€๊ฒฝ ์‹œ ์ด๋ฏธ์ง€ ์ƒ์„ฑ ํ•จ์ˆ˜ ํ˜ธ์ถœ
37
+ script_output.change(fn=script_to_images, inputs=script_output, outputs=images_output)
38
+
39
+ demo.launch()