Spaces:
No application file
No application file
| from fastapi import FastAPI | |
| from fastapi.openapi.utils import get_openapi | |
| from pydantic import BaseModel | |
| from classifier import Classifier | |
| class SentimentText(BaseModel): | |
| text: str | |
| app = FastAPI() | |
| def root(request: SentimentText): | |
| classfier = Classifier() | |
| prediction = classfier.classify(request.text) | |
| return {"sentiment": prediction} | |
| def endpoint_openapi(): | |
| if app.openapi_schema: | |
| return app.openapi_schema | |
| openapi_schema = get_openapi( | |
| title="Sentiment Analysis", | |
| version="2.5.0", | |
| description="This is the OpenAPI schema for the app", | |
| routes=app.routes, | |
| ) | |
| openapi_schema["info"]["x-logo"] = { | |
| "url": "https://fastapi.tiangolo.com/img/logo-margin/logo-teal.png" | |
| } | |
| app.openapi_schema = openapi_schema | |
| return app.openapi_schema | |
| app.openapi = endpoint_openapi |