Khriis commited on
Commit
6e81e10
verified
1 Parent(s): 97783d1

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +4 -2
handler.py CHANGED
@@ -51,8 +51,10 @@ class EndpointHandler:
51
 
52
  # Question template (must match training)
53
  self.question_template = (
54
- "What caused the {emotion}?"
55
- )
 
 
56
 
57
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
58
  """
 
51
 
52
  # Question template (must match training)
53
  self.question_template = (
54
+ "Extract the exact short phrase (<= 8 words) from the target "
55
+ "utterance that most strongly signals the emotion {emotion}. "
56
+ "Return only a substring of the target utterance."
57
+ )
58
 
59
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
60
  """