Update app.py

app.py

@@ -1,33 +1,30 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-# 無料スペースで確実に動く日本語モデル
-model = pipeline(
+model_name = "cyberagent/open-calm-1b"
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+pipe = pipeline(
     "text-generation",
-    model="cyberagent/open-calm-3b",
-    max_new_tokens=200,
+    model=model,
+    tokenizer=tokenizer,
+    max_new_tokens=100,
     do_sample=True,
     temperature=0.7
 )
 
 def chat(history, user_input):
-    response = model(user_input)[0]["generated_text"]
+    response = pipe(user_input)[0]["generated_text"]
     history.append((user_input, response))
     return history, ""
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 日本語文章の要約・質問チャット（軽量・安定版）")
-    
+    gr.Markdown("## 日本語文章の要約・質問チャット（超軽量版）")
     chatbot = gr.Chatbot(height=500)
-    text_input = gr.Textbox(
-        label="ここに文章を貼り付けて質問してください",
-        placeholder="例：この文章を要約して、など"
-    )
+    text_input = gr.Textbox(label="文章を入力してください")
 
-    text_input.submit(
-        chat,
-        inputs=[chatbot, text_input],
-        outputs=[chatbot, text_input]
-    )
+    text_input.submit(chat, [chatbot, text_input], [chatbot, text_input])
 
 demo.launch()