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()
|