lawq / app.py
honey90's picture
Update app.py
d39674a verified
raw
history blame contribute delete
920 Bytes
# ν•„μš”ν•œ 라이브러리 μ„€μΉ˜ λͺ…λ Ή (ν„°λ―Έλ„μ΄λ‚˜ λͺ…λ Ή ν”„λ‘¬ν”„νŠΈμ—μ„œ μ‹€ν–‰):
# pip install transformers gradio
# TensorFlow λ˜λŠ” PyTorch 쀑 ν•˜λ‚˜λ₯Ό μ„€μΉ˜ν•˜μ„Έμš”. 예λ₯Ό λ“€μ–΄, PyTorch μ„€μΉ˜ λͺ…λ Ή:
# pip install torch torchvision torchaudio
# λ˜λŠ” TensorFlow μ„€μΉ˜ λͺ…λ Ή:
# pip install tensorflow
# 라이브러리 μž„ν¬νŠΈ
import gradio as gr
from transformers import pipeline
# ν…μŠ€νŠΈ 생성 νŒŒμ΄ν”„λΌμΈ 생성
chatbot_pipeline = pipeline('text-generation', model='gpt2')
# μ‚¬μš©μžμ˜ μ§ˆλ¬Έμ— λŒ€ν•œ 닡변을 μƒμ„±ν•˜λŠ” ν•¨μˆ˜
def answer_question(question):
responses = chatbot_pipeline(question, max_length=100)
return responses[0]['generated_text']
# Gradio μΈν„°νŽ˜μ΄μŠ€ 생성
interface = gr.Interface(fn=answer_question, inputs="text", outputs="text", title="κ°„λ‹¨ν•œ 챗봇")
# μΈν„°νŽ˜μ΄μŠ€ μ‹€ν–‰
if __name__ == "__main__":
interface.launch()