|
from pipeline import PreTrainedPipeline |
|
|
|
|
|
my_handler = PreTrainedPipeline(path=".") |
|
|
|
doc = [ |
|
"The Half Moon is a public house and music venue in Putney, London. It is one of the city's longest running live music venues, and has hosted live music every night since 1963.", |
|
"The pub is on the south side of the Lower Richmond road, in the London Borough of Wandsworth.", |
|
] |
|
PARAGRAPH_DELIMITER = "\n\n" |
|
doc_string = PARAGRAPH_DELIMITER.join(doc) |
|
|
|
|
|
payload = doc_string |
|
prediction = my_handler(payload) |
|
|
|
|
|
print(prediction) |
|
|
