Update app.py

app.py

@@ -3,18 +3,26 @@ import gradio as gr
 # モデルのロード
 model = gr.load("models/Miwa-Keita/zenz-v1-checkpoints")
 
-# カスタム関数を定義
+# 入力を調整する関数
 def preprocess_input(user_input):
     prefix = "\uEE00"  # 前に付与する文字列
     suffix = "\uEE01"  # 後ろに付与する文字列
     processed_input = prefix + user_input + suffix
     return model(processed_input)
 
+# 出力を調整する関数
+def postprocess_output(model_output):
+    suffix = "\uEE01"
+    # \uEE01の後の部分を抽出
+    if suffix in model_output:
+        return model_output.split(suffix)[1]
+    return model_output
+
 # インターフェースを定義
 iface = gr.Interface(
-    fn=preprocess_input,
-    inputs="text",
-    outputs="text",
+    fn=lambda x: postprocess_output(preprocess_input(x)),
+    inputs=gr.inputs.Textbox(label="変換する文字列（カタカナ）"),
+    outputs=gr.outputs.Textbox(label="変換結果"),
     title="ニューラルかな漢字変換モデルzenz-v1のデモ",
     description="変換したい文字列をカタカナを入力してください"
 )