Update app.py

app.py

@@ -30,6 +30,10 @@ print(f'MiniMind模型参数量: {sum(p.numel() for p in model.parameters() if p
 
 # 生成文本函数
 def generate_text(prompt, max_length=512, temperature=0.85, top_p=0.85, history_cnt=0):
+    # 如果输入为空，则返回提示信息
+    if not prompt.strip():
+        return "请输入您的问题或指令..."
+    
     # 设置随机种子
     setup_seed(random.randint(0, 2048))
     
@@ -255,7 +259,7 @@ with gr.Blocks(css=custom_css) as demo:
             <div>
                 <span class="logo-pulse">🧠</span>
                 <h1>MiniMind AI</h1>
-                <p>强大的自然语言处理模型，为您提供智能对话体验</p>
+                <p>基于开源项目minimind训练的模型，为您提供智能对话体验</p>
             </div>
             """)
         
@@ -348,6 +352,13 @@ with gr.Blocks(css=custom_css) as demo:
         outputs=output_text
     )
     
+    # 添加回车键触发生成回答的功能
+    input_text.submit(
+        fn=generate_text,
+        inputs=[input_text, max_length, temperature, top_p, history_cnt],
+        outputs=output_text
+    )
+    
     clear_btn.click(
         fn=clear_history,
         inputs=[],