Update app.py

app.py

@@ -23,7 +23,19 @@ def generate(prefix_text):
     return tokenizer.decode(gen_token[0])
 
 
-# テキストエリアと送信ボタンを追加
-with gr.Interface(generate) as demo:
-    demo.textbox("テキストを入力してください", label="入力")
-    demo.button("送信", label="送信")
+def main():
+    # テキストエリアを追加
+    text_input = gr.Textbox()
+
+    # 送信ボタンを追加
+    submit_button = gr.Button("送信")
+
+    # テキストエリアと送信ボタンを連携
+    submit_button.on_click(lambda: gr.show(generate(text_input.value), title="テキスト生成"))
+
+    # アプリを起動
+    gr.Interface(main).launch()
+
+
+if __name__ == "__main__":
+    main()