Spaces:

shankulpatel1
/

summarizer

Build error

Update app.py

b224adf verified almost 2 years ago

350 Bytes

import gradio as gr
import transformers
import torch
from transformers import pipeline

summarizer = pipeline("summarization", model="google-t5/t5-base", tokenizer="google-t5/t5-base", framework="pt")

def summarize(text):
    return summarizer(text, max_length=20)
    

demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
demo.launch()

	import gradio as gr
	import transformers
	import torch
	from transformers import pipeline

	summarizer = pipeline("summarization", model="google-t5/t5-base", tokenizer="google-t5/t5-base", framework="pt")

	def summarize(text):
	return summarizer(text, max_length=20)


	demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
	demo.launch()