Update app.py

app.py

@@ -1,20 +1,24 @@
 from transformers import pipeline
 import gradio as gr
+import os
 
-model_repository_name = "Kosuke-Yamada/t5_ad_generation"
-ad_pipeline = pipeline("text2text-generation", model_repository_name, no_repeat_ngram_size=1)
+hf_token = os.getenvs("HF_TOKEN")
+
+model_repository_name = "Kosuke-Yamada/t5_ad_generation2"
+ad_pipeline = pipeline("text2text-generation", model_repository_name, token=hf_token, no_repeat_ngram_size=1)
 
 def main(text):
     return ad_pipeline(text)[0]["generated_text"]
 
 with gr.Blocks() as demo:
-    with gr.Row():
-        with gr.Column():
-            gr.Markdown("# 広告生成デモ")
-
-            input_text = gr.components.Textbox(lines=1, label="キーワードを空白区切りで入力して！")
-            submit_button = gr.Button("Submit")
-            output_text = gr.components.Textbox(lines=1, label="生成広告")
+    with gr.Column():
+        gr.Markdown("# 広告生成デモ")
+        with gr.Row():
+            with gr.Column():
+                input_text = gr.components.Textbox(lines=1, label="キーワードを空白区切りで入力して！")
+                submit_button = gr.Button("Submit")
+            with gr.Column():
+                output_text = gr.components.Textbox(lines=1, label="生成広告")
 
     submit_button.click(main, inputs=[input_text], outputs=[output_text])
 demo.launch()