Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
@@ -304,6 +304,7 @@ class InferenceRunner(StateManager):
|
|
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()
|
|
|
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()
|