summarizer / app.py
jrsenobio's picture
update app.py
4ad0efe verified
raw
history blame contribute delete
641 Bytes
import os
import gradio as gr
from transformers import pipeline
# 1. Get the secret token you just saved
hf_token = os.getenv("HF_TOKEN")
# 2. Load the model WITH the token
# The 'token' argument proves you have permission to access the private model
pipe = pipeline(
"summarization",
model="jrsenobio/summarizer_model",
token=hf_token
)
def summarize(text):
if not text:
return "No text provided"
# Run the model
output = pipe(text, max_length=130, min_length=30)
return output[0]['summary_text']
demo = gr.Interface(
fn=summarize,
inputs="text",
outputs="text"
)
demo.launch()