import gradio as gr from transformers import pipeline def text_generation_model(input_text): model = pipeline("text-generation", model="facebook/galactica-6.7b") output_text = model(input_text) return output_text input_text_interface = gr.inputs.Textbox(placeholder="Enter text here") output_text_interface = gr.outputs.Textbox() interface = gr.Interface(fn=text_generation_model, inputs=input_text_interface, outputs=output_text_interface, title="Text Generation Model") interface.launch()