Spaces:
Sleeping
Sleeping
Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
|
@@ -304,7 +304,6 @@ class InferenceRunner(StateManager):
|
|
| 304 |
f'high-end GPUs, implementation will be explored in the future.')
|
| 305 |
|
| 306 |
if self.is_model_loaded:
|
| 307 |
-
st.write(type(st.session_state['kbvqa']))
|
| 308 |
free_gpu_resources()
|
| 309 |
st.session_state['loading_in_progress'] = False
|
| 310 |
self.update_prev_state()
|
|
|
|
| 304 |
f'high-end GPUs, implementation will be explored in the future.')
|
| 305 |
|
| 306 |
if self.is_model_loaded:
|
|
|
|
| 307 |
free_gpu_resources()
|
| 308 |
st.session_state['loading_in_progress'] = False
|
| 309 |
self.update_prev_state()
|