File size: 475 Bytes
acb1db0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from handler import EndpointHandler

# init handler
my_handler = EndpointHandler(path="pbotsaris/musicgen-small")

non_holiday_payload = {"inputs": "I am quite excited how this will turn out", "duration": 2}
# prepare sample payload

# test the handler
non_holiday_pred=my_handler(non_holiday_payload)

# show results
print("done")
print(non_holiday_pred)

# non_holiday_pred [{'label': 'joy', 'score': 0.9985942244529724}]
# holiday_payload [{'label': 'happy', 'score': 1}]