mafoaurelie's picture
Create app.py
baf0d49 verified
raw
history blame contribute delete
550 Bytes
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()