Miwa-Keita commited on
Commit
b169a72
1 Parent(s): b896c7a

Update to easily try conversion

Browse files
Files changed (1) hide show
  1. app.py +21 -1
app.py CHANGED
@@ -1,3 +1,23 @@
1
  import gradio as gr
2
 
3
- gr.load("models/Miwa-Keita/zenz-v1-checkpoints").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
 
3
+ # モデルのロード
4
+ model = gr.load("models/Miwa-Keita/zenz-v1-checkpoints")
5
+
6
+ # カスタム関数を定義
7
+ def preprocess_input(user_input):
8
+ prefix = "\uEE00" # 前に付与する文字列
9
+ suffix = "\uEE01" # 後ろに付与する文字列
10
+ processed_input = prefix + user_input + suffix
11
+ return model(processed_input)
12
+
13
+ # インターフェースを定義
14
+ iface = gr.Interface(
15
+ fn=preprocess_input,
16
+ inputs="text",
17
+ outputs="text",
18
+ title="ニューラルかな漢字変換モデルzenz-v1のデモ",
19
+ description="変換したい文字列をカタカナを入力してください"
20
+ )
21
+
22
+ # ローンチ
23
+ iface.launch()