Spaces:
Sleeping
Sleeping
Typo
Browse files
state.py
CHANGED
|
@@ -519,7 +519,7 @@ def run_analysis_loop(file_names, file_paths_dict):
|
|
| 519 |
|
| 520 |
# Pull pipeline and move to device for inference
|
| 521 |
pipeline = st.session_state.pipeline
|
| 522 |
-
|
| 523 |
for i, fname in enumerate(pending):
|
| 524 |
fpath = file_paths_dict.get(fname, "")
|
| 525 |
ext = fpath.lower()
|
|
|
|
| 519 |
|
| 520 |
# Pull pipeline and move to device for inference
|
| 521 |
pipeline = st.session_state.pipeline
|
| 522 |
+
pipeline.to(pipeline.device)
|
| 523 |
for i, fname in enumerate(pending):
|
| 524 |
fpath = file_paths_dict.get(fname, "")
|
| 525 |
ext = fpath.lower()
|