xizaoqu
commited on
Commit
·
a7ea928
1
Parent(s):
2769bd7
update
Browse files
app.py
CHANGED
@@ -25,6 +25,7 @@ from datetime import datetime
|
|
25 |
import spaces
|
26 |
from algorithms.worldmem import WorldMemMinecraft
|
27 |
from huggingface_hub import hf_hub_download
|
|
|
28 |
|
29 |
torch.set_float32_matmul_precision("high")
|
30 |
|
@@ -278,13 +279,11 @@ def generate(keys):
|
|
278 |
out_video = np.clip(out_video, a_min=0.0, a_max=1.0)
|
279 |
out_video = (out_video * 255).astype(np.uint8)
|
280 |
|
281 |
-
|
282 |
-
|
283 |
-
filename = f"outputs_gradio/{timestamp}.mp4"
|
284 |
-
save_video(out_video, filename)
|
285 |
|
286 |
input_history += keys
|
287 |
-
return out_video[-1],
|
288 |
|
289 |
def reset():
|
290 |
global memory_curr_frame
|
|
|
25 |
import spaces
|
26 |
from algorithms.worldmem import WorldMemMinecraft
|
27 |
from huggingface_hub import hf_hub_download
|
28 |
+
import tempfile
|
29 |
|
30 |
torch.set_float32_matmul_precision("high")
|
31 |
|
|
|
279 |
out_video = np.clip(out_video, a_min=0.0, a_max=1.0)
|
280 |
out_video = (out_video * 255).astype(np.uint8)
|
281 |
|
282 |
+
temporal_video_path = tempfile.NamedTemporaryFile(suffix='.mp4').name
|
283 |
+
save_video(out_video, temporal_video_path)
|
|
|
|
|
284 |
|
285 |
input_history += keys
|
286 |
+
return out_video[-1], temporal_video_path, input_history
|
287 |
|
288 |
def reset():
|
289 |
global memory_curr_frame
|