Update app.py

app.py

@@ -13,7 +13,7 @@ def load_model():
     global model_loaded, load_error_msg
     print(f"🚀 正在启动... 准备加载模型: {MODEL_ID}")
     try:
-        # 1. 强制修正 Config (保持之前的修复)
+        # 1. 强制修正 Config
         config = AutoConfig.from_pretrained(MODEL_ID)
         config.tie_word_embeddings = True 
         
@@ -58,18 +58,14 @@ def generate_text(prompt, max_len, temp):
         # 解码全部内容
         full_response = tokenizer.decode(outputs[0], skip_special_tokens=True)
         
-        # --- ✨ 优化显示逻辑 ✨ ---
-        # 1. 只提取【新增】的内容，不要把原来的 Prompt 再显示一遍
-        # 这样右边的框里就只有 "Paris."，看起来更清爽
+        # 结果清洗：去掉输入部分
         new_text = full_response[len(prompt):]
         
-        # 2. 尝试去掉最后那个没说完的半句话 (如果有换行符)
-        # 如果模型写了 "Paris.\nThe capital of US is", 我们只保留 "Paris."
+        # 去掉未完成的句子
         if "\n" in new_text.strip():
-             # 取第一行非空的内容
              lines = [line for line in new_text.split('\n') if line.strip()]
              if lines:
-                 return lines[0] # 只返回第一行
+                 return lines[0] 
         
         return new_text
         
@@ -77,7 +73,8 @@ def generate_text(prompt, max_len, temp):
         return str(e)
 
 # --- ✨ 全新 UI 布局 (Blocks) ✨ ---
-with gr.Blocks(theme="soft", title="MobileLLM 80M Demo") as demo:
+# ⚠️ 修复点：去掉了 theme 和 title 参数，防止报错
+with gr.Blocks() as demo:
     
     # 标题区域
     gr.Markdown(
@@ -93,24 +90,24 @@ with gr.Blocks(theme="soft", title="MobileLLM 80M Demo") as demo:
         with gr.Column():
             input_box = gr.Textbox(
                 label="输入 Prompt (提示词)",
-                lines=8,  # 高度设大一点
+                lines=10, 
                 placeholder="在这里输入排比句...",
                 value="The capital of China is Beijing.\nThe capital of Japan is Tokyo.\nThe capital of Germany is Berlin.\nThe capital of France is"
             )
             
-            # 高级参数折叠区
-            with gr.Accordion("⚙️ 高级参数 (点我展开)", open=False):
+            # 高级参数
+            with gr.Accordion("⚙️ 高级参数", open=False):
                 slider_len = gr.Slider(minimum=1, maximum=100, value=20, label="生成长度", step=1)
                 slider_temp = gr.Slider(minimum=0.1, maximum=1.0, value=0.6, label="温度 (创造力)", step=0.1)
             
-            submit_btn = gr.Button("🚀 开始生成", variant="primary", scale=1)
+            submit_btn = gr.Button("🚀 开始生成", variant="primary")
 
         # 右边：输出区
         with gr.Column():
             output_box = gr.Textbox(
                 label="模型续写结果",
-                lines=8, # 高度设大一点，防止显示不全
-                interactive=False # 输出框不可编辑
+                lines=10, 
+                interactive=False 
             )
 
     # 绑定点击事件