app.py

import gradio as gr
from transformers import pipeline

# Load the model from Hugging Face
generator = pipeline('text-generation', model='JuanCabs/lapepav1')

# Define the function for text generation
def generate_text(prompt):
    results = generator(prompt, max_length=100)
    return results[0]['generated_text']

# Create the Gradio interface
interface = gr.Interface(
    fn=generate_text,  # The function to execute
    inputs="text",     # Input type
    outputs="text",    # Output type
    title="Text Generation Model",  # Title of the app
    description="Enter a prompt to generate text using a Hugging Face model"  # App description
)

# Launch the app
interface.launch()