File size: 347 Bytes
5f6a25a 4258af5 5f6a25a 4258af5 5f6a25a 4258af5 5f6a25a 4258af5 5f6a25a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from fastapi import FastAPI
from transformers import pipeline
app = FastAPI()
pipe = pipeline("text-classification", model="FacebookAI/roberta-large-mnli")
# @app.get("/")
# def home():
# return{"message":"hello rath"}
@app.get("/genrate")
def gentxt(text : str):
output = pipe(text)
return({"output":output[0]["generated_text"]}) |