JUNGU's picture
Update app.py
c7367c6
raw
history blame contribute delete
769 Bytes
from transformers import pipeline
import random
import gradio as gr
from translate import Translator
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM
def classfi(input):
classifier= pipeline("sentiment-analysis")
translator= Translator(from_lang="ko", to_lang="en")
inputko=translator.translate(input)
return classifier(inputko)
with gr.Blocks(css=".gradio-container {background-color: white}") as demo:
name=gr.Textbox(label="ํ•˜๊ณ  ์‹ถ์€ ์ด์•ผ๊ธฐ๋ฅผ ์ ์–ด ๊ฐ์ • ๋ถ„์„ํ•˜๊ธฐ ๋ฒ„ํŠผ์„ ๋ˆŒ๋Ÿฌ๋ณด์„ธ์š”")
output=gr.Textbox(label="๋‹น์‹ ์˜ ๊ฐ์ • ์ƒํƒœ๋ฅผ ์•Œ๋ ค์ค๋‹ˆ๋‹ค.")
greet_btn = gr.Button("๊ฐ์ • ๋ถ„์„ํ•˜๊ธฐ")
greet_btn.click(fn=classfi, inputs=name, outputs=output)
demo.launch(debug = True, share =False)