app.py

import gradio as gr
from transformers import pipeline

# Load the model directly from Hugging Face Model Hub
generator = pipeline("text-generation", model="LiquidAI/LFM2-1.2B")

# Function to generate text from input prompt
def generate_text(prompt):
    result = generator(prompt, max_length=500, num_return_sequences=1)
    return result[0]['generated_text']

# Create a Gradio interface
iface = gr.Interface(fn=generate_text,
                     inputs="text",
                     outputs="text",
                     title="GPT-2 Text Generation",
                     description="Enter a prompt and see the text generated by GPT-2!")

# Launch the interface
iface.launch()