Spaces:
Runtime error
Runtime error
da03
commited on
Commit
·
9ced953
1
Parent(s):
6c00a2b
main.py
CHANGED
|
@@ -136,6 +136,7 @@ def process_frame(
|
|
| 136 |
# Decoding
|
| 137 |
start = time.perf_counter()
|
| 138 |
sample = sample_latent * DATA_NORMALIZATION['std'] + DATA_NORMALIZATION['mean']
|
|
|
|
| 139 |
sample = model.decode_first_stage(sample)
|
| 140 |
sample = sample.squeeze(0).clamp(-1, 1)
|
| 141 |
timing['decode'] = time.perf_counter() - start
|
|
@@ -220,7 +221,7 @@ async def websocket_endpoint(websocket: WebSocket):
|
|
| 220 |
|
| 221 |
# Print global FPS measurement
|
| 222 |
print(f" Global FPS: {global_fps:.2f} (total: {frame_count} frames in {total_elapsed:.2f}s)")
|
| 223 |
-
|
| 224 |
|
| 225 |
img = Image.fromarray(sample_img)
|
| 226 |
buffered = io.BytesIO()
|
|
|
|
| 136 |
# Decoding
|
| 137 |
start = time.perf_counter()
|
| 138 |
sample = sample_latent * DATA_NORMALIZATION['std'] + DATA_NORMALIZATION['mean']
|
| 139 |
+
time.sleep(10)
|
| 140 |
sample = model.decode_first_stage(sample)
|
| 141 |
sample = sample.squeeze(0).clamp(-1, 1)
|
| 142 |
timing['decode'] = time.perf_counter() - start
|
|
|
|
| 221 |
|
| 222 |
# Print global FPS measurement
|
| 223 |
print(f" Global FPS: {global_fps:.2f} (total: {frame_count} frames in {total_elapsed:.2f}s)")
|
| 224 |
+
|
| 225 |
|
| 226 |
img = Image.fromarray(sample_img)
|
| 227 |
buffered = io.BytesIO()
|