aeprod / app.py
NoNeuronsNoStress's picture
Update app.py
a87d8e5
import gradio
import gradio as gr
import os
from transformers import pipeline
summarizer = pipeline("summarization")
def my_inference_function(text,key: str):
if len(str(text)) < 10:
return "Error, text is too short!"
if key != os.environ.get("key"):
return "Error, the key is incorrect!"
return summarizer(text)
gradio_interface = gradio.Interface(
fn = my_inference_function,
inputs =[gr.Textbox(
label="prompt", placeholder="What is the capital of France"),
gr.Textbox(label="key")],
outputs = "text"
)
gradio_interface.launch()