Update app.py

app.py

@@ -9,18 +9,33 @@ from attributtes_utils import input_pose, input_emotion, input_blink
 
 
 @spaces.GPU
+
 def process(input_vid, audio_path, pose_select, emotion_select, blink_select):
     model = init_model()
     pose = input_pose(pose_select)
     emotion = input_emotion(emotion_select)
-    os.system(f"ffmpeg -y -loglevel error -i {audio_path} -vn -acodec pcm_s16le -ar 16000 -ac 1 2_output.wav")
+    
+    # Convert audio with error handling
+    print(audio_path, input_vid)
+    result = os.system(f"ffmpeg -y -loglevel error -i {audio_path} -vn -acodec pcm_s16le -ar 16000 -ac 1 2_output.wav")
+    if result != 0:
+        raise RuntimeError("Failed to execute ffmpeg command. Please check the input audio file.")
+    
+    # Check if output file exists
+    if not os.path.exists("2_output.wav"):
+        raise FileNotFoundError("2_output.wav was not created. Check the ffmpeg command and input file.")
+    
     blink = input_blink(blink_select)
-
     print("input_vid: ", input_vid)
-    result = infenrece(model, input_vid, f"2_output.wav", pose, emotion, blink)
+    
+    # Perform inference
+    try:
+        result = infenrece(model, input_vid, "2_output.wav", pose, emotion, blink)
+    except Exception as e:
+        raise RuntimeError(f"Inference failed: {e}")
+    
     print("result: ", result)
-
-    print("finished !")
+    print("finished!")
 
     return result  # , gr.Group.update(visible=True)