Avinaash's picture
Creating app.py
f7efddf
raw
history blame contribute delete
559 Bytes
import gradio
from transformers import pipeline
def model_pipeline(input):
model = pipeline(model='{git_name}')
try:
result = model(input)
except Exception as e:
result = f"Model couldn't process input, refer to model card at https://huggingface.co/{git_name}.)"
return result
gradio_interface = gradio.Interface(
fn=model_pipeline,
inputs="text",
outputs="json",
title=f"Your personal space for {git_name_clean}",
description=f"Model card:+ https://huggingface.co/{git_name}",
)
gradio_interface.launch()