Gijs Wijngaard
commited on
Commit
·
4dd3faa
1
Parent(s):
cf6397c
Refine inference instructions for concise segments
Browse files- 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 = "
|
| 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 |
|