chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,29 @@
|
||||
import requests
|
||||
from starlette.requests import Request
|
||||
from typing import Dict
|
||||
|
||||
from transformers import pipeline
|
||||
|
||||
from ray import serve
|
||||
|
||||
|
||||
# 1: Wrap the pretrained sentiment analysis model in a Serve deployment.
|
||||
@serve.deployment
|
||||
class SentimentAnalysisDeployment:
|
||||
def __init__(self):
|
||||
self._model = pipeline("sentiment-analysis")
|
||||
|
||||
def __call__(self, request: Request) -> Dict:
|
||||
return self._model(request.query_params["text"])[0]
|
||||
|
||||
|
||||
# 2: Deploy the deployment.
|
||||
serve.run(SentimentAnalysisDeployment.bind(), route_prefix="/")
|
||||
|
||||
# 3: Query the deployment and print the result.
|
||||
print(
|
||||
requests.get(
|
||||
"http://localhost:8000/", params={"text": "Ray Serve is great!"}
|
||||
).json()
|
||||
)
|
||||
# {'label': 'POSITIVE', 'score': 0.9998476505279541}
|
||||
Reference in New Issue
Block a user