Spaces:

M-Rizwan0099
/

Text-2-text-with-dockers

Sleeping

Upload 3 files

baa15a5 verified almost 2 years ago

506 Bytes

from fastapi import FastAPI
from transformers import pipeline


##create a new FASTAPI app
app = FastAPI()

## Initialize the pipeline for text generation
pipe = pipeline("text2text-generation", model="google/flan-t5-small")

@app.get("/")
def home():
    return {"message": "Welcome to the Text Generation API"}

@app.get("/generate_text/")
def generate_text(text: str):
    result = pipe(text)
    
    ##return the generated text in json format
    return {"generated_text": result[0]["generated_text"]}

	from fastapi import FastAPI
	from transformers import pipeline


	##create a new FASTAPI app
	app = FastAPI()

	## Initialize the pipeline for text generation
	pipe = pipeline("text2text-generation", model="google/flan-t5-small")

	@app.get("/")
	def home():
	return {"message": "Welcome to the Text Generation API"}

	@app.get("/generate_text/")
	def generate_text(text: str):
	result = pipe(text)

	##return the generated text in json format
	return {"generated_text": result[0]["generated_text"]}