Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
|
@@ -116,6 +116,7 @@ class InferenceRunner(StateManager):
|
|
| 116 |
self.image_qa_app(self.get_model())
|
| 117 |
|
| 118 |
if load_fine_tuned_model:
|
|
|
|
| 119 |
free_gpu_resources()
|
| 120 |
self.load_model()
|
| 121 |
|
|
|
|
| 116 |
self.image_qa_app(self.get_model())
|
| 117 |
|
| 118 |
if load_fine_tuned_model:
|
| 119 |
+
st.write("TEXTS")
|
| 120 |
free_gpu_resources()
|
| 121 |
self.load_model()
|
| 122 |
|