Hugging Face's logo

Spaces:
m7mdal7aj
/
KB-VQA
Sleeping

App Files Community
2
m7mdal7aj commited on
Commit
d7c0f13
·
verified ·
1 Parent(s): a264403

Update my_model/KBVQA.py

Browse files
Files changed (1) hide show
  1. my_model/KBVQA.py +2 -1
my_model/KBVQA.py CHANGED
@@ -185,7 +185,8 @@ class KBVQA:
185
  free_gpu_resources()
186
 
187
  free_gpu_resources()
188
- prepare_kbvqa_model(only_reload_detection_model=False, force_reload=True)
 
189
 
190
 
191
  def format_prompt(self, current_query: str, history: Optional[str] = None, sys_prompt: Optional[str] = None, caption: str = None, objects: Optional[str] = None) -> str:
 
185
  free_gpu_resources()
186
 
187
  free_gpu_resources()
188
+
189
+ return prepare_kbvqa_model(only_reload_detection_model=False, force_reload=True)
190
 
191
 
192
  def format_prompt(self, current_query: str, history: Optional[str] = None, sys_prompt: Optional[str] = None, caption: str = None, objects: Optional[str] = None) -> str: