app.py

import os
import gradio as gr
from transformers import pipeline

def generate_text(prompt):
    generator = pipeline('text-generation', model='tiiuae/falcon-40b-instruct', trust_remote_code=True)
    return generator(prompt, max_length=100)[0]['generated_text']

def main():
    interface = gr.Interface(
        fn=generate_text,
        inputs=gr.inputs.Textbox(lines=2, placeholder="Type your prompt here..."),
        outputs="text",
        title="Text Generation Model",
        description="This model generates text based on the input prompt. Powered by Hugging Face transformers."
    )
    
    interface.launch()

if __name__ == "__main__":
    main()