Spaces:
Build error
Build error
import gradio as gr | |
from transformers import pipeline | |
pipeline = pipeline(task="text-generation", model="Dilana/Babu_panchi") | |
def generate(input_text): | |
return pipeline(input_text) | |
iface = gr.Interface(fn = generate, | |
inputs = "text", | |
outputs = ['text'], | |
title = 'text generation', | |
description = "Get well structure prompt from given input") | |
iface.launch(inline = False) |