Spaces:
Runtime error
Runtime error
Update gradio_app.py
Browse files- gradio_app.py +3 -3
gradio_app.py
CHANGED
@@ -258,9 +258,9 @@ def get_image(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, i
|
|
258 |
save_videos(batch_samples, result_dir, filenames=[prompt_str], fps=save_fps)
|
259 |
print(f"Saved in {prompt_str}. Time used: {(time.time() - start):.2f} seconds")
|
260 |
model = model.cpu()
|
261 |
-
|
262 |
-
print("result saved to:",
|
263 |
-
return
|
264 |
|
265 |
|
266 |
# @spaces.GPU
|
|
|
258 |
save_videos(batch_samples, result_dir, filenames=[prompt_str], fps=save_fps)
|
259 |
print(f"Saved in {prompt_str}. Time used: {(time.time() - start):.2f} seconds")
|
260 |
model = model.cpu()
|
261 |
+
saved_result_dir = os.path.join(result_dir, f"{prompt_str}.mp4")
|
262 |
+
print("result saved to:", saved_result_dir)
|
263 |
+
return saved_result_dir
|
264 |
|
265 |
|
266 |
# @spaces.GPU
|