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()
|