mdsr commited on
Commit
7bf8c01
·
1 Parent(s): caf38b9

using `mp4v` video codec

Browse files
Files changed (3) hide show
  1. app.py +9 -5
  2. packages.txt +1 -1
  3. requirements.txt +1 -1
app.py CHANGED
@@ -5,21 +5,25 @@ model = slt.models.ConcatenativeSynthesis("ur", "pk-sl", "video")
5
 
6
 
7
  def text_to_video(
8
- text: str, text_language: str, sign_language: str, output_path: str = "output.mp4"
 
 
 
 
9
  ):
10
  model.text_language = text_language
11
  model.sign_language = sign_language
12
 
13
  video = model.translate(text)
14
- video.save(output_path, overwrite=True)
15
 
16
  # ToDo: video.watermark("Sign Language Translator\nAI Generated Video")
17
 
18
 
19
  def predict(text: str, text_language: str, sign_language: str):
20
  try:
21
- path = "./output.mkv"
22
- text_to_video(text, text_language, sign_language, output_path=path)
23
  return path
24
  except Exception as exc:
25
  return gradio.Error(f"Error during translation: {exc}")
@@ -45,7 +49,7 @@ gradio_app = gradio.Interface(
45
  ),
46
  ], # type: ignore
47
  outputs=gradio.Video(
48
- format="mkv",
49
  label="Synthesized Sign Language Video",
50
  autoplay=True,
51
  show_download_button=True,
 
5
 
6
 
7
  def text_to_video(
8
+ text: str,
9
+ text_language: str,
10
+ sign_language: str,
11
+ output_path: str = "output.mp4",
12
+ codec="h264",
13
  ):
14
  model.text_language = text_language
15
  model.sign_language = sign_language
16
 
17
  video = model.translate(text)
18
+ video.save(output_path, overwrite=True, codec=codec)
19
 
20
  # ToDo: video.watermark("Sign Language Translator\nAI Generated Video")
21
 
22
 
23
  def predict(text: str, text_language: str, sign_language: str):
24
  try:
25
+ path = "./output.mp4"
26
+ text_to_video(text, text_language, sign_language, output_path=path, codec="mp4v")
27
  return path
28
  except Exception as exc:
29
  return gradio.Error(f"Error during translation: {exc}")
 
49
  ),
50
  ], # type: ignore
51
  outputs=gradio.Video(
52
+ format="mp4",
53
  label="Synthesized Sign Language Video",
54
  autoplay=True,
55
  show_download_button=True,
packages.txt CHANGED
@@ -1,4 +1,4 @@
1
- ffmpeg
2
  x264
3
  libx264-dev
 
4
  python3-opencv
 
 
1
  x264
2
  libx264-dev
3
+ ffmpeg
4
  python3-opencv
requirements.txt CHANGED
@@ -1,2 +1,2 @@
1
  sign-language-translator
2
- opencv-python
 
1
  sign-language-translator
2
+ opencv-python==4.1.1.26