Rename 01_B&W_Videos_Colorizer.py to 01_黑白影视着色器.py

01_B&W_Videos_Colorizer.py → 01_黑白影视着色器.py

@@ -14,28 +14,28 @@ from tqdm import tqdm
 from models.deep_colorization.colorizers import eccv16
 from utils import load_lottieurl, format_time, colorize_frame, change_model
 
-st.title("B&W Videos Colorizer")
+st.title("黑白影视着色器")
 
 st.write("""
-    ##### Upload a black and white video and get a colorized version of it.
-    ###### ➠ This space is using CPU Basic so it might take a while to colorize a video.""")
+    ##### 上传黑白视频并获得其彩色版本。
+    ###### ➠ 空间使用的是 CPU Basic，因此对视频进行彩色化可能需要一些时间。""")
 
-#st.set_page_config(page_title="Image & Video Colorizer", page_icon="🎨", layout="wide")
+#st.set_page_config(page_title="黑白影视着色器", page_icon="🎨", layout="wide")
 
 loaded_model = eccv16(pretrained=True).eval()
 current_model = "None"
 
 def main():
     model = st.selectbox(
-        "Select Model (Both models have their pros and cons, I recommend trying both and keeping the best for your task)",
+        "选择模型（两种模型都有其优点和缺点，建议尝试两种模型并保持最适合您的任务）",
         ["ECCV16", "SIGGRAPH17"], index=0)
 
     loaded_model = change_model(current_model, model)
-    st.write(f"Model is now {model}")
+    st.write(f"现在是 {model}")
 
-    uploaded_file = st.file_uploader("Upload your video here...", type=['mp4', 'mov', 'avi', 'mkv'])
+    uploaded_file = st.file_uploader("在这里上传您的视频...", type=['mp4', 'mov', 'avi', 'mkv'])
 
-    if st.button("Colorize"):
+    if st.button("着色"):
         if uploaded_file is not None:
             file_extension = os.path.splitext(uploaded_file.name)[1].lower()
             if file_extension in ['.mp4', '.avi', '.mov', '.mkv']:
@@ -53,20 +53,20 @@ def main():
 
                 col1, col2 = st.columns([0.5, 0.5])
                 with col1:
-                    st.markdown('<p style="text-align: center;">Before</p>', unsafe_allow_html=True)
+                    st.markdown('<p style="text-align: center;">之前</p>', unsafe_allow_html=True)
                     st.video(temp_file.name)
 
                 with col2:
-                    st.markdown('<p style="text-align: center;">After</p>', unsafe_allow_html=True)
+                    st.markdown('<p style="text-align: center;">之后</p>', unsafe_allow_html=True)
 
-                    with st.spinner("Colorizing frames..."):
+                    with st.spinner("上色中..."):
                         # Colorize video frames and store in a list
                         output_frames = []
                         total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
                         progress_bar = st.progress(0)  # Create a progress bar
 
                         start_time = time.time()
-                        time_text = st.text("Time Remaining: ")  # Initialize text value
+                        time_text = st.text("剩余时间：")  # Initialize text value
 
                         for _ in tqdm(range(total_frames), unit='frame', desc="Progress"):
                             ret, frame = video.read()
@@ -89,7 +89,7 @@ def main():
                                 time_text.empty()  # Remove text value
                                 progress_bar.empty()
 
-                    with st.spinner("Merging frames to video..."):
+                    with st.spinner("将帧合并到视频..."):
                         frame_size = output_frames[0].shape[:2]
                         output_filename = "output.mp4"
                         fourcc = cv2.VideoWriter_fourcc(*"mp4v")  # Codec for MP4 video
@@ -116,7 +116,7 @@ def main():
 
                         # Add a download button for the colorized video
                         st.download_button(
-                            label="Download Colorized Video",
+                            label="下载彩色视频",
                             data=open(converted_filename, "rb").read(),
                             file_name="colorized_video.mp4"
                         )