Spaces:
Runtime error
Runtime error
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() | |