bugfix

src/predict.py

@@ -255,9 +255,9 @@ class Predictor:
             final_output = str(openvoice_output).replace('wav',output_extension)
             self.convert_wav_to_output_extension(openvoice_output,final_output)
         
-        self.upload_file_to_s3(final_output,'demovidelyusergenerations',f"{gen_id}-voice-clone.{output_extension}")
+        self.upload_file_to_s3(final_output,bucket_name,f"{gen_id}-voice-clone.{output_extension}")
         shutil.rmtree(os.path.join(output_dir,gen_id))
-        return {"voice_clone":f"https://demovidelyusergenerations.s3.amazonaws.com/{gen_id}-voice-clone.{output_extension}"
+        return {"voice_clone":f"https://{bucket_name}.s3.amazonaws.com/{gen_id}-voice-clone.{output_extension}"
                 }
             
     
@@ -298,9 +298,9 @@ class Predictor:
             sf.write(openvoice_output,wav1,new_sr)
         final_ouput = str(openvoice_output).replace('wav',output_extension)
         self.convert_wav_to_output_extension(openvoice_output,final_ouput)
-        self.upload_file_to_s3(final_ouput,'demovidelyusergenerations',f"{gen_id}-voice-clone-emotions.{output_extension}")
+        self.upload_file_to_s3(final_ouput,bucket_name,f"{gen_id}-voice-clone-emotions.{output_extension}")
         shutil.rmtree(os.path.join(output_dir,gen_id))
-        return {"voice_clone_emotions":f"https://demovidelyusergenerations.s3.amazonaws.com/{gen_id}-voice-clone-emotions.{output_extension}",
+        return {"voice_clone_emotions":f"https://{bucket_name}.s3.amazonaws.com/{gen_id}-voice-clone-emotions.{output_extension}",
                 }
         
 
@@ -349,9 +349,9 @@ class Predictor:
         final_output = str(multi_lang_with_voice_clone_path).replace('wav',output_extension)
         self.convert_wav_to_output_extension(multi_lang_with_voice_clone_path,final_output)
         print(final_output)
-        self.upload_file_to_s3(final_output,'demovidelyusergenerations',f"{gen_id}-voice-clone-multi-lang.{output_extension}")
+        self.upload_file_to_s3(final_output,bucket_name,f"{gen_id}-voice-clone-multi-lang.{output_extension}")
         shutil.rmtree(os.path.join(output_dir,gen_id))
-        return {"voice_clone_with_multi_lang":f"https://demovidelyusergenerations.s3.amazonaws.com/{gen_id}-voice-clone-multi-lang.{output_extension}"
+        return {"voice_clone_with_multi_lang":f"https://{bucket_name}.s3.amazonaws.com/{gen_id}-voice-clone-multi-lang.{output_extension}"
         }