Update app.py

app.py

@@ -23,7 +23,7 @@ def video_to_frames(video_path, output_folder):
         cv2.imwrite(frame_name, image)
         success, image = vidcap.read()
         count += 1
-        if count>500:
+        if count>550:
           break
     print(f"{count} frames extracted from {video_path}.")
     return [count,fps]
@@ -68,8 +68,7 @@ def frames_to_video(frame_folder, video_path, image_path, frame_count,fps):
         else:
             out.write(img1)
         progress = int((i + 1) / frame_count * 100)
-        if progress%5==0 or progress%4==0:
-            print(progress,end="")
+        print(progress,end="")
     out.release()
     
     print(f"Video saved at {video_path}.")
@@ -77,8 +76,8 @@ def frames_to_video(frame_folder, video_path, image_path, frame_count,fps):
 def face_swap(video_path, image_path):
     output_folder = "Out_Frames"
     frame_count = video_to_frames(video_path, output_folder)
-    if frame_count[0] > 150:
-        frame_count[0] = 150
+    if frame_count[0] > 500:
+        frame_count[0] = 500
     output_video_path = "output_video.mp4" 
     frames_to_video(output_folder, output_video_path, image_path, frame_count[0],frame_count[1])
     return output_video_path 
@@ -88,7 +87,7 @@ iface = gr.Interface(
     fn=face_swap,
     inputs=["video", "image"],
     outputs="video",
-    title="Face Swap",
+    title="Profaker's Face Swap",
     description="Upload a video and an image. The faces in the video will be swapped with the face in the image.",
 )
 iface.launch(share=True)