SamuelMiller/sum_it

1# import libraries2import torch3from transformers import pipeline4import gradio as gr5from transformers import Speech2TextProcessor, Speech2TextForConditionalGeneration6from gradio.mix import Parallel, Series7 8# Display Text9desc = "Summarize your text!"10 11# Models I've fine tuned12qa_model = 'huggingface/SamuelMiller/qa_squad'13my_model_1 = 'huggingface/SamuelMiller/lil_sumsum'14my_model_2 = 'huggingface/SamuelMiller/lil_sum_sum'15my_model_3 = 'huggingface/SamuelMiller/sum_sum'16 17# Currently using this model for demo, from https://huggingface.co/google/pegasus-large18better_model = 'huggingface/google/pegasus-large'19 20# Summarize Function21def summarize(text):22  summ = gr.Interface.load(better_model)      23  summary = summ(text)24  return summary25  26# Gradio Interface  27iface = gr.Interface(fn=summarize,28                 theme='huggingface', 29                 title= 'sum_it', 30                 description= desc,31                 inputs= 'textbox',32                 outputs= 'textbox')33                 34# Launch Interface                 35iface.launch(inline = False)36 37