arxivgpt kim commited on
Commit
80d88d2
ยท
verified ยท
1 Parent(s): 882b8a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -15
app.py CHANGED
@@ -1,18 +1,30 @@
 
 
 
 
1
  import gradio as gr
2
 
3
- def create_content(purpose, audience, tone, word_count, style, format_type):
4
- # ๋‚ด์šฉ ์ฐฝ์ž‘ ๋กœ์ง์„ ์—ฌ๊ธฐ์— ๊ตฌํ˜„ํ•ฉ๋‹ˆ๋‹ค. ์˜ˆ๋ฅผ ๋“ค์–ด, ์ž…๋ ฅ๋ฐ›์€ ์ •๋ณด๋ฅผ ๋ฐ”ํƒ•์œผ๋กœ ์ ์ ˆํ•œ ๋‚ด์šฉ์„ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
5
- content = f"์—ฌ๊ธฐ์— ์ƒ์„ฑ๋œ ๋‚ด์šฉ์ด ํ‘œ์‹œ๋ฉ๋‹ˆ๋‹ค. ๋ชฉ์ : {purpose}, ๋Œ€์ƒ ์ฒญ์ค‘: {audience}, ํ†ค: {tone}, ๋‹จ์–ด ์ˆ˜: {word_count}, ์Šคํƒ€์ผ: {style}, ํ˜•์‹: {format_type}"
6
- return content
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
 
8
- with gr.Interface(fn=create_content,
9
- inputs=[gr.Textbox(label="๋ชฉ์ "),
10
- gr.Textbox(label="๋Œ€์ƒ ์ฒญ์ค‘"),
11
- gr.Dropdown(choices=["๊ณต์‹์ ์ธ", "๋น„๊ณต์‹์ ์ธ", "์„ค๋“๋ ฅ ์žˆ๋Š”"], label="ํ†ค"),
12
- gr.Number(label="๋‹จ์–ด ์ˆ˜"),
13
- gr.Textbox(label="์Šคํƒ€์ผ"),
14
- gr.Dropdown(choices=["๋ธ”๋กœ๊ทธ ํฌ์ŠคํŠธ", "๋ณด๋„ ์ž๋ฃŒ", "๊ธฐ์‚ฌ"], label="์ฝ˜ํ…์ธ  ํ˜•์‹")],
15
- outputs="text",
16
- title="Write For Me - ์ฝ˜ํ…์ธ  ์ƒ์„ฑ๊ธฐ",
17
- description="์‚ฌ์šฉ์ž์˜ ์š”๊ตฌ์‚ฌํ•ญ์— ๋งž์ถฐ ์ฝ˜ํ…์ธ ๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.") as interface:
18
- interface.launch()
 
1
+ # ํ•„์š”ํ•œ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋ฅผ ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค
2
+ # pip install gradio transformers
3
+
4
+ from transformers import pipeline
5
  import gradio as gr
6
 
7
+ # GPT ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•˜๋Š” ํŒŒ์ดํ”„๋ผ์ธ์„ ์ดˆ๊ธฐํ™”ํ•ฉ๋‹ˆ๋‹ค. ๋ชจ๋ธ ์‹๋ณ„์ž๋Š” ์‚ฌ์šฉํ•˜๊ณ ์ž ํ•˜๋Š” GPT ๋ชจ๋ธ์— ๋”ฐ๋ผ ๋‹ฌ๋ผ์ง‘๋‹ˆ๋‹ค.
8
+ # ์˜ˆ: 'gpt-2', 'EleutherAI/gpt-neo-2.7B' ๋“ฑ. GPT-4๋ฅผ ์‚ฌ์šฉํ•  ๊ฒฝ์šฐ ํ•ด๋‹น ๋ชจ๋ธ์˜ ์ •ํ™•ํ•œ ์‹๋ณ„์ž๋ฅผ ์‚ฌ์šฉํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค.
9
+ text_generator = pipeline('text-generation', model='gpt-4-model-identifier')
10
+
11
+ def generate_text(prompt, max_length=50, temperature=0.7):
12
+ # ํ•จ์ˆ˜๋Š” ์ฃผ์–ด์ง„ ํ”„๋กฌํ”„ํŠธ๋ฅผ ๋ฐ”ํƒ•์œผ๋กœ ํ…์ŠคํŠธ๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
13
+ generated_texts = text_generator(prompt, max_length=max_length, temperature=temperature, return_full_text=False)
14
+ return generated_texts[0]['generated_text']
15
+
16
+ # Gradio ์ธํ„ฐํŽ˜์ด์Šค๋ฅผ ๊ตฌ์„ฑํ•ฉ๋‹ˆ๋‹ค.
17
+ interface = gr.Interface(
18
+ fn=generate_text, # ํ˜ธ์ถœํ•  ํ•จ์ˆ˜
19
+ inputs=[ # ์‚ฌ์šฉ์ž๋กœ๋ถ€ํ„ฐ ์ž…๋ ฅ๋ฐ›์„ ์š”์†Œ๋“ค
20
+ gr.inputs.Textbox(lines=5, label="Input Text"),
21
+ gr.inputs.Slider(minimum=10, maximum=100, default=50, label="Maximum Length"),
22
+ gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=0.7, label="Temperature")
23
+ ],
24
+ outputs=gr.outputs.Textbox(label="Generated Text"), # ๊ฒฐ๊ณผ๋ฅผ ํ‘œ์‹œํ•  ์š”์†Œ
25
+ title="GPT-4 Text Generator", # ์ธํ„ฐํŽ˜์ด์Šค์˜ ์ œ๋ชฉ
26
+ description="Enter text to see how GPT-4 can expand, summarize, or rewrite it." # ์ธํ„ฐํŽ˜์ด์Šค์˜ ์„ค๋ช…
27
+ )
28
 
29
+ # ์ธํ„ฐํŽ˜์ด์Šค๋ฅผ ์‹คํ–‰ํ•ฉ๋‹ˆ๋‹ค.
30
+ interface.launch()