write / app.py
arxivgpt kim
Update app.py
80d88d2 verified
# ํ•„์š”ํ•œ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋ฅผ ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค
# 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()