Update app.py

app.py

@@ -41,7 +41,7 @@ def main():
 
     with c2:
         st.caption("")
-        st.title("MaMaL-Gen(代码生成)")
+        st.title("MaMaL-Gen(代码补全)")
 
 
     ############ SIDEBAR CONTENT ############
@@ -94,21 +94,23 @@ def main():
     """
     )
 
-    generator = pipeline('text-generation', model="hanbin/MaMaL-Com")
+   
 
 
 
     # model, tokenizer = load_model("hanbin/MaMaL-Gen")
     st.write("### 输入：")
     input = st.text_area("",height=200)
+    button = st.button('补全')
+    generator = pipeline('text-generation', model="hanbin/MaMaL-Com")
     output = generator(input)
     # code = '''def hello():
     #     print("Hello, Streamlit!")'''
-    if st.button('补全'):
+    if button:
         st.write("### 输出：")
         st.code(output, language='python')
     else:
-        st.write('')
+        st.write('请等待输出，如果长时间未输出，可能是您网络不通畅。也可能是我们并没有花钱购买服务器，用的是免费的服务器。您可以下载模型到本地测试~')