Update app.py

app.py

@@ -8,7 +8,7 @@ from transformers import pipeline, set_seed
 set_seed(42)
 
 # 从环境变量获取 token（现在应该可以正常工作了）
-HF_TOKEN = os.environ.get('ProtTex')
+HF_TOKEN = os.environ.get('HF_TOKEN')
 print(f"Token 状态: {'已设置' if HF_TOKEN else '未设置'}")
 
 # 初始化模型管道
@@ -113,16 +113,9 @@ def generate_protein(instruction, max_length=100):
     except Exception as e:
         yield f"❌ 生成过程中出现未预期错误: {str(e)}"
 
-# 创建界面
+# 创建界面 - 修复了 theme 参数问题
 with gr.Blocks(
-    title="ProtTeX 蛋白质生成器",
-    theme=gr.themes.Soft(),
-    css="""
-    .gradio-container {
-        max-width: 900px;
-        margin: auto;
-    }
-    """
+    title="ProtTeX 蛋白质生成器"
 ) as demo:
     
     gr.Markdown("""
@@ -160,8 +153,7 @@ with gr.Blocks(
             output = gr.Textbox(
                 label="🧬 生成的蛋白质序列",
                 lines=8,
-                show_copy_button=True,
-                elem_id="output-box"
+                show_copy_button=True
             )
     
     # 示例部分