Spaces:

iamAI123
/

podsum

Sleeping

iamAI123 commited on Jul 15, 2023

Commit

a0fc304

1 Parent(s): 2099820

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,11 +2,11 @@ from transformers import pipeline
 import gradio as gr
 def get_summary(text):
-    summarizer = pipeline("summarization", model="gmurro/bart-large-finetuned-filtered-spotify-podcast-summ", tokenizer="gmurro/bart-large-finetuned-filtered-spotify-podcast-summ")
     summary = summarizer(text, min_length=39, max_length=25000)
     return summary[0]['summary_text']
 #gr.Interface(fn=get_summary, inputs="text",  outputs="text").launch()
-gr.Interface.load("huggingface/gpt2").launch()

 import gradio as gr
 def get_summary(text):
+    summarizer = pipeline("summarization", model="paulowoicho/t5-podcast-summarisation", tokenizer="paulowoicho/t5-podcast-summarisation")
     summary = summarizer(text, min_length=39, max_length=25000)
     return summary[0]['summary_text']
 #gr.Interface(fn=get_summary, inputs="text",  outputs="text").launch()
+gr.Interface.load("paulowoicho/t5-podcast-summarisation").launch()