import gradio as gr | |
import transformers | |
from transformers import pipeline | |
generator=pipeline('text-generation',model='microsoft/phi-2') | |
generator('Hello i am an artificial intelligence engineer',max_length=40) | |
def Get_GenerationText(input_text): | |
result= generator(input_text) | |
return result[0]['generated_text'] | |
iface=gr.Interface(fn=Get_GenerationText, | |
title='Generation Text', | |
inputs='text', | |
outputs='text', | |
description='Get Text Generation of giving input').launch() |