my_translator / app.py
soarhigh's picture
Upload app.py
b3b38b5
# -*- coding: utf-8 -*-
"""demo_gradio_huggingface_spaces.ipynb
Automatically generated by Colaboratory.
Original file is located at
https://colab.research.google.com/drive/1vy72Qs-scdzH68lhYbGN11fLzcE9Md1O
"""
#!pip install transformers gradio sentencepiece
#!pip install transformers[sentencepiece]
#!pip install sacremoses
# gradio ๊ทธ๋ผ๋””์˜ค ๊ทธ๋ ˆ์ด๋””์˜ค
# !๋Š” ์ฝ”๋žฉ์—๊ฒŒ ํ•˜๋Š” ์‹œ์Šคํ…œ ๋ช…๋ น์–ด
# ์‹œ์Šคํ…œ ๋ช…๋ น์–ด?๊ฐ€ ๋ญ์ฃ : ํ‚ค์›Œ๋“œ -> ์œ ๋‹‰์Šค ์‰˜ ์Šคํฌ๋ฆฝํŠธ ๋ช…๋ น์–ด
# ! ์—†์ด๋Š” ๊ทธ๋ƒฅ ์ผ๋ฐ˜ ํŒŒ์ด์ฌ ๋ช…๋ น์–ด
# sklearn/pandas/plotly๋Š” ์„ค์น˜ ์•ˆ ํ–ˆ๋Š”๋ฐ?
# ์ฝ”๋žฉ ๋จธ์‹ ์— ๊ธฐ๋ณธ์ ์œผ๋กœ ์„ค์น˜๋˜์–ด์žˆ๋Š”๋ฐ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋“ค์ด ์žˆ์Œ
import gradio as gr # gradio๋ฅผ ๊ฐ€์ ธ์™€์„œ ์“ฐ๋Š”๋ฐ, ๋งค๋ฒˆ gradio ์—ฌ์„ฏ์ž ์น˜๋Š”๊ฒŒ ๊ท€์ฐฎ์œผ๋ฉด
# as gr์ด๋ผ๊ณ  ์ถ”๊ฐ€ํ•ด์„œ ์ด ์ดํ›„๋กœ๋Š” gr์ด๋ผ๊ณ ๋งŒ ์จ๋„ ๋จ
# gradio๊ฐ€ ์›น ์ƒ์—์„œ์˜ ๋ฐ๋ชจ๋ฅผ ๋งŒ๋“ค์–ด์คŒ (UI ํฌํ•จ!)
import transformers # ํŠธ๋žœ์Šคํฌ๋จธ(์‹ ๊ฒฝ๋ง์˜ ์ผ์ข…)
from transformers import pipeline # ํŒŒ์ดํ”„๋ผ์ธ์€ ๋จธ์‹ ๋Ÿฌ๋‹ ๋ฌธ์ œ๋ฅผ ํ•ด๊ฒฐํ•˜๋Š” ๋ฐ ํ•„์š”ํ•œ
# ์—ฌ๋Ÿฌ ๊ตฌ์„ฑ ์š”์†Œ๋ฅผ ๋ฌถ์–ด์คŒ -> ๊ตฌ์„ฑ ์š”์†Œ๋ž€? ์˜ˆ) ๋ฌธ์ œ์˜ ์ข…๋ฅ˜, ๋ชจ๋ธ์˜ ์ข…๋ฅ˜ ๋“ฑ
pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
# ๋ฌธ์ œ์˜ ์ข…๋ฅ˜๋Š” ๋ฒˆ์—ญ์ด๊ณ , ์šฐ๋ฆฌ๊ฐ€ ์“ธ ๋ชจ๋ธ์€ Hel.... en์ด๋ผ ๋ถˆ๋ฆฌ๋Š” ๋ชจ๋ธ
demo = gr.Interface.from_pipeline(pipe)
# gradio๋ผ๋Š” ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ์— ์ •์˜๋˜์–ด์žˆ๋Š” Interface๋ฅผ ๋งŒ๋“ค์ž
# ๋ญ๋กœ๋ถ€ํ„ฐ ๋งŒ๋“œ๋ƒ๋ฉด pipe๋กœ๋ถ€ํ„ฐ
# UI -> User Interface
demo.launch()
# [email protected]