Gijs Wijngaard commited on
Commit
4dd3faa
·
1 Parent(s): cf6397c

Refine inference instructions for concise segments

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -22,7 +22,7 @@ from transformers import Qwen2_5OmniProcessor
22
 
23
  ADAPTER_PATH = (Path(__file__).resolve().parent / "model" / "checkpoint-22000").resolve()
24
  DEFAULT_SYSTEM_PROMPT = "You are an audio segmentation assistant. For each turn, return JSON with the requested keys."
25
- DEFAULT_INSTRUCTION = "Provide a JSON object with start_time, end_time, and caption fields describing the first salient event in the clip."
26
  USE_AUDIO_IN_VIDEO = False
27
 
28
 
 
22
 
23
  ADAPTER_PATH = (Path(__file__).resolve().parent / "model" / "checkpoint-22000").resolve()
24
  DEFAULT_SYSTEM_PROMPT = "You are an audio segmentation assistant. For each turn, return JSON with the requested keys."
25
+ DEFAULT_INSTRUCTION = "General note: The segments should be relatively concise (1-8 seconds). To begin, respond with a JSON object containing `start_time` and `end_time` for the initial segment. Return the JSON only, with no extra commentary."
26
  USE_AUDIO_IN_VIDEO = False
27
 
28