Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -196,9 +196,6 @@ class KBVQA():
         Returns:
             str: The formatted prompt for the KBVQA model.
         """
-
-        if sys_prompt is None:
-            sys_prompt = config.SYSTEM_PROMPT 
     
         B_SENT = '<s>'
         E_SENT = '</s>'
@@ -212,10 +209,9 @@ class KBVQA():
         E_QES = '[/QES]'
         B_OBJ = '[OBJ]'
         E_OBJ = '[/OBJ]'
-    
         current_query = current_query.strip()
-        sys_prompt = sys_prompt.strip()
-        
+        if sys_prompt is None:
+            sys_prompt = config.SYSTEM_PROMPT.strip()
         if history is None:
             if objects is None:
                 p = f"""{B_SENT}{B_INST} {B_SYS}{sys_prompt}{E_SYS}{B_CAP}{caption}{E_CAP}{B_QES}{current_query}{E_QES}{E_INST}"""