Update hf_gradio_app.py

hf_gradio_app.py

@@ -85,8 +85,8 @@ def process_audio(file_path, temp_dir):
     # Load the audio file
     audio = AudioSegment.from_file(file_path)
 
-    # Check and cut the audio if longer than 4 seconds
-    max_duration = 9 * 1000  # 4 seconds in milliseconds
+    # Check and cut the audio if longer than 9 seconds
+    max_duration = 9 * 1000  # 9 seconds in milliseconds
     if len(audio) > max_duration:
         audio = audio[:max_duration]
 
@@ -99,7 +99,7 @@ def process_audio(file_path, temp_dir):
     return output_path
 
 
-@spaces.GPU(duration=540)
+@spaces.GPU(duration=580)
 @torch.inference_mode()
 def generate(input_video, input_audio, seed, progress=gr.Progress(track_tqdm=True)):
     """
@@ -116,7 +116,7 @@ def generate(input_video, input_audio, seed, progress=gr.Progress(track_tqdm=Tru
         str: File path to the generated output video (MP4 format).
     """
 
-    gr.Info("540 seconds will be allocated from your daily ZeroGPU time credits.")
+    gr.Info("580 seconds will be allocated from your daily ZeroGPU time credits.")
     
     pipeline.reference_net.enable_xformers_memory_efficient_attention()
     pipeline.diffusion_net.enable_xformers_memory_efficient_attention()
@@ -225,25 +225,13 @@ def generate(input_video, input_audio, seed, progress=gr.Progress(track_tqdm=Tru
 with gr.Blocks(analytics_enabled=False) as demo:
     with gr.Column():
         gr.Markdown("# MEMO: Memory-Guided Diffusion for Expressive Talking Video Generation")
-        gr.Markdown("Note: On fffiloni's shared UI, audio length is trimmed to max 4 seconds, so everyone can get a taste without to much waiting time in queue.")
-        gr.Markdown("Duplicate the space to skip the queue and enjoy full length capacity.")
+        gr.Markdown("Note: Audio length is trimmed to max 9 seconds, using 580 GPU credits.")
+        gr.Markdown("Consider adding word at beginning of clip that is later trimmable since lipsync does not always start immediately.")
         gr.HTML("""
         <div style="display:flex;column-gap:4px;">
             <a href="https://github.com/memoavatar/memo">
                 <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
             </a>
-            <a href="https://memoavatar.github.io/">
-                <img src='https://img.shields.io/badge/Project-Page-green'>
-            </a>
-			<a href="https://arxiv.org/abs/2412.04448">
-                <img src='https://img.shields.io/badge/ArXiv-Paper-red'>
-            </a>
-            <a href="https://huggingface.co/spaces/fffiloni/MEMO?duplicate=true">
-				<img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-sm.svg" alt="Duplicate this Space">
-			</a>
-			<a href="https://huggingface.co/fffiloni">
-				<img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/follow-me-on-HF-sm-dark.svg" alt="Follow me on HF">
-			</a>
         </div>
         """)