Update app.py

app.py

@@ -6,7 +6,9 @@ import cv2
 from huggingface_hub import InferenceClient
 import torch
 
-device = torch.device("cpu")
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+print('Using {} for inference.'.format(device))
+
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 def generate_output(prompt):
@@ -27,6 +29,7 @@ def generate_output(prompt):
     video_path = "video.mp4"
     command = f"CUDA_VISIBLE_DEVICES='' python3 inference.py --checkpoint_path checkpoints/wav2lip_gan.pth --face face.jpg --audio {audio_path} --outfile {video_path} --nosmooth --resize_factor 2"
     process = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+    
     if process.returncode != 0:
         error_message = process.stderr
         return None, f"No se pudo generar el video: {error_message}"
@@ -43,5 +46,5 @@ if st.button("Generar Video"):
     if error_message:
         st.error(f"Error: {error_message}")
     else:
-        video_file = open(video_path, "rb").read()
-        st.video(video_file)
+        with open(video_path, "rb") as video_file:
+            st.video(video_file.read())