vivaceailab commited on
Commit
a05f62d
Β·
verified Β·
1 Parent(s): 4cf3686

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -4
app.py CHANGED
@@ -1,7 +1,26 @@
1
  import gradio as gr
 
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
5
 
6
- demo = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- demo.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
 
4
+ # Hugging Face의 Helsinki-NLP Ko->Ru λ²ˆμ—­κΈ° μ‚¬μš©
5
+ translator = pipeline("translation_ko_to_ru", model="Helsinki-NLP/opus-mt-ko-ru")
6
 
7
+ def chat_translate(user_input):
8
+ if not user_input.strip():
9
+ return "μž…λ ₯을 ν•΄μ£Όμ„Έμš”!"
10
+
11
+ # λ²ˆμ—­
12
+ result = translator(user_input)
13
+ translated_text = result[0]['translation_text']
14
+
15
+ return translated_text
16
+
17
+ # κ°„λ‹¨ν•œ Gradio μ±— μΈν„°νŽ˜μ΄μŠ€
18
+ iface = gr.ChatInterface(
19
+ fn=chat_translate,
20
+ chatbot_name="ν•œκ΅­μ–΄->λŸ¬μ‹œμ•„μ–΄ λ²ˆμ—­ 챗봇",
21
+ title="ν•œκ΅­μ–΄ μž…λ ₯μ‹œ λŸ¬μ‹œμ•„μ–΄ λ²ˆμ—­ 챗봇",
22
+ description="ν•œκ΅­μ–΄λ‘œ μ§ˆλ¬Έν•˜λ©΄ λŸ¬μ‹œμ•„μ–΄λ‘œ λ²ˆμ—­ν•΄λ“œλ¦½λ‹ˆλ‹€."
23
+ )
24
+
25
+ if __name__ == "__main__":
26
+ iface.launch()