Spaces:

Kuberwastaken
/

TREAT

Running

Kuberwastaken commited on Jan 9

Commit

9c86082

1 Parent(s): e1c6460

Fixing Runtime Error

Files changed (1) hide show

model/analyzer.py CHANGED Viewed

@@ -96,7 +96,7 @@ class ContentAnalyzer:
         for chunk_idx, chunk in enumerate(script_chunks, 1):
             print(f"\n--- Processing Chunk {chunk_idx}/{len(script_chunks)} ---")
             for category, info in trigger_categories.items():
-                score, response = self.analyze_chunk(chunk, info)
                 if response == "YES":
                     identified_triggers[category] = identified_triggers.get(category, 0) + 1
                 elif response == "MAYBE":

         for chunk_idx, chunk in enumerate(script_chunks, 1):
             print(f"\n--- Processing Chunk {chunk_idx}/{len(script_chunks)} ---")
             for category, info in trigger_categories.items():
+                _, response = self.analyze_chunk(chunk, info)
                 if response == "YES":
                     identified_triggers[category] = identified_triggers.get(category, 0) + 1
                 elif response == "MAYBE":