Spaces:

kulvir63
/

text-generation

Sleeping

text-generation / app.py

Update app.py

3f8f39c verified about 2 months ago

No virus

276 Bytes

	from fastapi import FastAPI


	from transformers import pipeline

	app = FastAPI()

	pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")

	@app.get("/infer_t5")
	def t5(input):
	output = pipe_flan(input)
	return {"output": output[0]["generated_text"]}