File size: 391 Bytes
ac34d1e a96e8ad ac34d1e 7e396f3 ac34d1e |
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("text2text-generation", model="nvidia/Nemotron-Research-Reasoning-Qwen-1.5B")
@app.get("/")
def home():
return {"message": "Welcome to the text generation API!"}
@app.get("/generate")
def generate_text(text: str):
output=pipe(text)
return {"output": output[0]['generated_text']} |