close files before remove

This commit is contained in:
Believethehype 2024-01-15 10:58:16 +01:00
parent 692eb6f38b
commit dafdd3d9d8
2 changed files with 4 additions and 2 deletions

View File

@ -91,9 +91,11 @@ def check_server_status(jobID, address) -> str | pd.DataFrame:
elif content_type == 'video/mp4':
with open('./outputs/video.mp4', 'wb') as f:
f.write(response.content)
f.close()
clip = VideoFileClip("./outputs/video.mp4")
clip.write_videofile("./outputs/video2.mp4")
result = upload_media_to_hoster("./outputs/video2.mp4")
clip.close()
os.remove("./outputs/video.mp4")
os.remove("./outputs/video2.mp4")
return result

View File

@ -40,7 +40,7 @@ class VideoGenerationSVD(DVMTaskInterface):
request_form["trainerFilePath"] = r'modules\stablevideodiffusion\stablevideodiffusion.trainer'
url = ""
frames = 14 # 25
frames = 7 # 25
model = "stabilityai/stable-video-diffusion-img2vid-xt" #,stabilityai/stable-video-diffusion-img2vid
@ -68,7 +68,7 @@ class VideoGenerationSVD(DVMTaskInterface):
options = {
"model": model,
"fps": 14
"fps": frames
}
request_form['options'] = json.dumps(options)