Dilana's picture
Create app.py
98f4712 verified
raw
history blame contribute delete
No virus
456 Bytes
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)