dasemu/Transcriptarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update subgen.py

Browse Source
This commit is contained in:
McCloudS
2023-01-28 12:03:24 -07:00
committed by GitHub
parent 6e098fac64
commit 6f75b6ccf3
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
subgen/subgen.py
View File

@@ -41,10 +41,13 @@ def process_post_request(request, *args, **kwargs):
print("event: " + event) print("event: " + event)
if ((procaddedmedia and event == "added") or (procmediaonplay and event == "played")) and (len(glob.glob("{}/{}*subgen*".format(filepath, filenamenoextension))) == 0): #glob nonsense checks if there exists a subgen file already and won't make a new one if ((procaddedmedia and event == "added") or (procmediaonplay and event == "played")) and (len(glob.glob("{}/{}*subgen*".format(filepath, filenamenoextension))) == 0): #glob nonsense checks if there exists a subgen file already and won't make a new one
if whisper_speedup: if os.getenv('WHISPER_SPEEDUP') == "True" :
print("This is a speedup run!")
print(os.getenv('WHISPER_SPEEDUP'))
finalsubname = "{0}/{1}.subgen.{2}.speedup.{3}".format( finalsubname = "{0}/{1}.subgen.{2}.speedup.{3}".format(
filepath, filenamenoextension, whisper_model, namesublang) filepath, filenamenoextension, whisper_model, namesublang)
else: else:
print("No speedup")
finalsubname = "{0}/{1}.subgen.{2}.{3}".format( finalsubname = "{0}/{1}.subgen.{2}.{3}".format(
filepath, filenamenoextension, whisper_model, namesublang) filepath, filenamenoextension, whisper_model, namesublang)
@@ -84,7 +87,7 @@ if not os.path.isdir("/whisper.cpp"):
os.mkdir("/whisper.cpp") os.mkdir("/whisper.cpp")
os.chdir("/whisper.cpp") os.chdir("/whisper.cpp")
subprocess.call("git clone https://github.com/ggerganov/whisper.cpp .", shell=True) subprocess.call("git clone https://github.com/ggerganov/whisper.cpp .", shell=True)
if updaterepo: if os.getenv('UPDATEREPO') == "True":
print("Updating repo!") print("Updating repo!")
subprocess.call("git pull", shell=True) subprocess.call("git pull", shell=True)
if os.path.isfile("/whisper.cpp/samples/jfk.wav"): if os.path.isfile("/whisper.cpp/samples/jfk.wav"):