sum_it / scratch.py
SamuelMiller's picture
Create scratch.py
0fad229
raw
history blame
No virus
2.03 kB
# >>>>>> Adapted/frankensteined from these scripts: <<<<<<<
# for Summary Interface:
# >>>> https://huggingface.co/spaces/khxu/pegasus-text-summarizers/blob/main/app.py
# Audio Interface
# >>>> https://huggingface.co/spaces/iSky/Speech-audio-to-text-with-grammar-correction/blob/main/app.py
# Gramar
# >>>> https://huggingface.co/deep-learning-analytics/GrammarCorrector/blob/main/README.md
import gradio as gr
from transformers import pipeline
from gradio.mix import Parallel, Series
# >>>>>>>>>>>>>>>>>>>> Danger Below <<<<<<<<<<<<<<<<<<<<<<
# Load Interfaces:
s2t = gr.Interface.load('huggingface/hf-internal-testing/processor_with_lm')
grammar = gr.Interface.load('huggingface/deep-learning-analytics/GrammarCorrector')
sum_it = gr.Interface.load('huggingface/SamuelMiller/lil_sum_sum')
# Audio Functions:
def out(audio):
flag = True
if audio==None:
return "no audio"
elif flag:
a = s2t(audio)
#g = grammar(a)
#s = sum_it(g) # Summarize Audio with sum_it
return a #grammar(a, num_return_sequences=1) # grammar(s), # Grammar Filter
else:
return "something is wrong in the function?"
# Construct Interfaces:
iface = gr.Interface(
fn=out,
title="Speech Audio to text (with corrected grammar)",
description="Let's Hear It!! This app transforms your speech (input) to text with corrected grammar after (output)!",
inputs= gr.inputs.Audio(source="microphone", type="filepath", label=None, optional=True),
outputs= 'text'
)
# Launch Interface
iface.launch(enable_queue=True,show_error=True)
# From Original Code:
# gr.inputs.Audio(source="upload", type="filepath", label=None, optional=True),
# examples=[["Grammar-Correct-Sample.mp3"], ["Grammar-Wrong-Sample.mp3"],],
#def speech_to_text(inp):
#pass # speech recognition model defined here
#gr.Interface(speech_to_text, inputs="mic", outputs=gr.Textbox(label="Predicted text", lines=4))