File size: 1,545 Bytes
80d88d2
 
 
 
882b8a0
 
80d88d2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
882b8a0
80d88d2
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# ํ•„์š”ํ•œ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋ฅผ ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค
# pip install gradio transformers

from transformers import pipeline
import gradio as gr

# GPT ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•˜๋Š” ํŒŒ์ดํ”„๋ผ์ธ์„ ์ดˆ๊ธฐํ™”ํ•ฉ๋‹ˆ๋‹ค. ๋ชจ๋ธ ์‹๋ณ„์ž๋Š” ์‚ฌ์šฉํ•˜๊ณ ์ž ํ•˜๋Š” GPT ๋ชจ๋ธ์— ๋”ฐ๋ผ ๋‹ฌ๋ผ์ง‘๋‹ˆ๋‹ค.
# ์˜ˆ: 'gpt-2', 'EleutherAI/gpt-neo-2.7B' ๋“ฑ. GPT-4๋ฅผ ์‚ฌ์šฉํ•  ๊ฒฝ์šฐ ํ•ด๋‹น ๋ชจ๋ธ์˜ ์ •ํ™•ํ•œ ์‹๋ณ„์ž๋ฅผ ์‚ฌ์šฉํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค.
text_generator = pipeline('text-generation', model='gpt-4-model-identifier')

def generate_text(prompt, max_length=50, temperature=0.7):
    # ํ•จ์ˆ˜๋Š” ์ฃผ์–ด์ง„ ํ”„๋กฌํ”„ํŠธ๋ฅผ ๋ฐ”ํƒ•์œผ๋กœ ํ…์ŠคํŠธ๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
    generated_texts = text_generator(prompt, max_length=max_length, temperature=temperature, return_full_text=False)
    return generated_texts[0]['generated_text']

# Gradio ์ธํ„ฐํŽ˜์ด์Šค๋ฅผ ๊ตฌ์„ฑํ•ฉ๋‹ˆ๋‹ค.
interface = gr.Interface(
    fn=generate_text,  # ํ˜ธ์ถœํ•  ํ•จ์ˆ˜
    inputs=[  # ์‚ฌ์šฉ์ž๋กœ๋ถ€ํ„ฐ ์ž…๋ ฅ๋ฐ›์„ ์š”์†Œ๋“ค
        gr.inputs.Textbox(lines=5, label="Input Text"),
        gr.inputs.Slider(minimum=10, maximum=100, default=50, label="Maximum Length"),
        gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=0.7, label="Temperature")
    ],
    outputs=gr.outputs.Textbox(label="Generated Text"),  # ๊ฒฐ๊ณผ๋ฅผ ํ‘œ์‹œํ•  ์š”์†Œ
    title="GPT-4 Text Generator",  # ์ธํ„ฐํŽ˜์ด์Šค์˜ ์ œ๋ชฉ
    description="Enter text to see how GPT-4 can expand, summarize, or rewrite it."  # ์ธํ„ฐํŽ˜์ด์Šค์˜ ์„ค๋ช…
)

# ์ธํ„ฐํŽ˜์ด์Šค๋ฅผ ์‹คํ–‰ํ•ฉ๋‹ˆ๋‹ค.
interface.launch()