Hugging Face's logo

Spaces:
m7mdal7aj
/
KB-VQA
Sleeping

App Files Community
2
m7mdal7aj commited on
Commit
c45c1b1
·
verified ·
1 Parent(s): 904c909

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +1 -1
my_model/tabs/run_inference.py CHANGED
@@ -115,7 +115,7 @@ class InferenceRunner(StateManager):
115
  free_gpu_resources()
116
  self.image_qa_app(self.get_model())
117
 
118
- elif load_fine_tuned_model:
119
  free_gpu_resources()
120
  self.load_model()
121
 
 
115
  free_gpu_resources()
116
  self.image_qa_app(self.get_model())
117
 
118
+ if load_fine_tuned_model:
119
  free_gpu_resources()
120
  self.load_model()
121