Update pages/03_B&W_Images_Colorizer.py

pages/03_B&W_Images_Colorizer.py

@@ -10,36 +10,36 @@ from utils import colorize_image, change_model, load_lottieurl
 
 #st.set_page_config(page_title="Image & Video Colorizer", page_icon="🎨", layout="wide")
 
-st.title("B&W Images Colorizer")
+st.title("黑白图像着色器")
 
 
 loaded_model = eccv16(pretrained=True).eval()
 current_model = "None"
 
 st.write("""
-    ##### Input a black and white image and get a colorized version of it.
-    ###### ➠ If you want to colorize multiple images just upload them all at once.
-    ###### ➠ Uploading already colored images won't raise errors but images won't look good.""")
+    ##### 输入黑白图像并获得其彩色版本。
+    ###### ➠ 如果您想对多张图像进行着色，只需一次上传它们即可。
+    ###### ➠ 上传已经着色的图像不会引发错误，但图像看起来不太好。""")
 
 
 def main():
     model = st.selectbox(
-        "Select Model (Both models have their pros and cons, I recommend trying both and keeping the best for you task)",
+        "选择模型（两种模型都有其优点和缺点，我建议尝试两种模型并保持最适合您的任务）",
         ["ECCV16", "SIGGRAPH17"], index=0)
 
     # Make the user select a model
     loaded_model = change_model(current_model, model)
-    st.write(f"Model is now {model}")
+    st.write(f"现在是 {model}")
 
     # Ask the user if he wants to see colorization
-    display_results = st.checkbox('Display results in real time', value=True)
+    display_results = st.checkbox('实时显示结果', value=True)
 
     # Input for the user to upload images
-    uploaded_file = st.file_uploader("Upload your images here...", type=['jpg', 'png', 'jpeg'],
+    uploaded_file = st.file_uploader("在这里上传您的图片...", type=['jpg', 'png', 'jpeg'],
                                      accept_multiple_files=True)
 
     # If the user clicks on the button
-    if st.button("Colorize"):
+    if st.button("着色"):
         # If the user uploaded images
         if uploaded_file is not None:
             if display_results:
@@ -59,7 +59,7 @@ def main():
                         with col1:
                             st.image(image, use_column_width="always")
                         with col2:
-                            with st.spinner("Colorizing image..."):
+                            with st.spinner("为图像着色..."):
                                 out_img, new_img = colorize_image(file, loaded_model)
                                 new_img.save("IMG_" + str(i+1) + ".jpg")
                                 st.image(out_img, use_column_width="always")
@@ -78,20 +78,20 @@ def main():
                 with col2:
                     # Provide the zip file data for download
                     st.download_button(
-                        label="Download Colorized Images" if len(uploaded_file) > 1 else "Download Colorized Image",
+                        label="下载彩色图像" if len(uploaded_file) > 1 else "下载彩色图像",
                         data=open(zip_filename, "rb").read(),
                         file_name=zip_filename,
                     )
             else:
                 with col2:
                     st.download_button(
-                        label="Download Colorized Image",
+                        label="下载彩色图像",
                         data=open("IMG_1.jpg", "rb").read(),
                         file_name="IMG_1.jpg",
                     )
 
         else:
-            st.warning('Upload a file', icon="⚠️")
+            st.warning('请上传文件', icon="⚠️")
 
 
 if __name__ == "__main__":