Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
|
@@ -164,8 +164,7 @@ class InferenceRunner(StateManager):
|
|
| 164 |
|
| 165 |
|
| 166 |
if self.is_model_loaded():
|
| 167 |
-
|
| 168 |
-
st.write(f'time is {time.time()-t1}')
|
| 169 |
free_gpu_resources()
|
| 170 |
st.session_state['loading_in_progress'] = False
|
| 171 |
self.image_qa_app(self.get_model())
|
|
|
|
| 164 |
|
| 165 |
|
| 166 |
if self.is_model_loaded():
|
| 167 |
+
|
|
|
|
| 168 |
free_gpu_resources()
|
| 169 |
st.session_state['loading_in_progress'] = False
|
| 170 |
self.image_qa_app(self.get_model())
|