arshadrana/Third_project

1from transformers import pipeline2from fastapi import FastAPI3from pydantic import BaseModel4 5app = FastAPI()6 7# Load the text generation pipeline from Hugging Face8generator = pipeline('text-generation', model='gpt2')9 10# Define the request body11class TextInput(BaseModel):12    input_text: str13 14@app.post("/generate")15def generate_text(text_input: TextInput):16    # Generate text from the model based on the input17    generated = generator(text_input.input_text, max_length=150, num_return_sequences=1)18    return {"generated_text": generated[0]['generated_text']}19