Update app.py

app.py

@@ -425,7 +425,7 @@ def generate_video(image, prompt, max_tokens):
         # Overlay ảnh heatmap lên ảnh gốc
         overlay = cv2.addWeighted(image, 1 - alpha, heatmap_color, alpha, 0)
         
-        prev_text = generated_text[:-len(input_token)-len(predict_token_text)] + "   "
+        prev_text = generated_text[:-len(input_token)-len(predict_token_text)] 
         params_for_text.append((prev_text, input_token, predict_token_text))
     
         hidden_tabel = extract_next_token_table_data(model, tokenizer, generation_output, index_focus)
@@ -476,7 +476,7 @@ def generate_video(image, prompt, max_tokens):
     video = VideoFileClip("heatmap_animation.mp4")
     audio = AudioFileClip("legacy-of-the-century-background-cinematic-music-for-video-46-second-319542.mp3").set_duration(video.duration)
     final = video.set_audio(audio)
-    final.write_videofile("heatmap_with_music.mp4", codec="libx264", audio_codec="aac")
+    final.write_videofile("heatmap_with_music.mp4", codec="libx264", audio_codec="aac", ffmpeg_params=["-pix_fmt", "yuv420p"])
     
     return "heatmap_with_music.mp4"