Upload llm_preprocess.py

llm_preprocess.py

@@ -86,9 +86,7 @@ def preprocess_gemini(free_text_caption):
     ]
 
     timestampCaption = model.generate_content(prompt_parts).text
-    #timestampCaption = model.generate_content(["explain what is dog bark"])
-    import pdb;pdb.set_trace()
-    # output = "dog_barking at 0.562-2.562_4.25-6.25_7.01-8.21."
+    
     return postprocess(timestampCaption)
 
 def preprocess_gpt(free_text_caption):
@@ -106,7 +104,7 @@ def preprocess_gpt(free_text_caption):
                 )
 
     timestampCaption = completion_start.choices[0].message.content
-    #output = "dog_barking at 0.562-2.562_4.25-6.25_7.01-8.21."
+    
     return postprocess(timestampCaption)
 
 if __name__=="__main__":