KB-VQA

Running

m7mdal7aj commited on Feb 28

Commit

c45c1b1

•

1 Parent(s): 904c909

Update my_model/tabs/run_inference.py

Files changed (1) hide show

my_model/tabs/run_inference.py CHANGED Viewed

@@ -115,7 +115,7 @@ class InferenceRunner(StateManager):
                         free_gpu_resources()
                         self.image_qa_app(self.get_model())
-                elif load_fine_tuned_model:
                     free_gpu_resources()
                     self.load_model()

                         free_gpu_resources()
                         self.image_qa_app(self.get_model())
+                if load_fine_tuned_model:
                     free_gpu_resources()
                     self.load_model()