Spaces:
Runtime error
Runtime error
jpdiazpardo
commited on
Commit
•
03978da
1
Parent(s):
258ccff
Update app.py
Browse files
app.py
CHANGED
@@ -46,14 +46,14 @@ title = "Scream: Fine-Tuned Whisper model for automatic gutural speech recogniti
|
|
46 |
classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", top_k=None)
|
47 |
|
48 |
#Functions-----------------------------------------------------------------------------------------------------------------------
|
49 |
-
def transcribe(
|
50 |
'''inputs: file, return_timestamps'''
|
51 |
-
outputs = pipe(
|
52 |
text = outputs["text"]
|
53 |
timestamps = outputs["chunks"]
|
54 |
|
55 |
#If return timestamps is True, return html text with timestamps format
|
56 |
-
if
|
57 |
spider_text = [f"{chunk['text']}" for chunk in timestamps] #Text for spider chart without timestamps
|
58 |
timestamps = [f"[{format_timestamp(chunk['timestamp'][0])} -> {format_timestamp(chunk['timestamp'][1])}] {chunk['text']}" for chunk in timestamps]
|
59 |
|
@@ -69,7 +69,7 @@ def transcribe(*args):
|
|
69 |
av_dict = calculate_average(trans_dict)
|
70 |
fig = spider_chart(av_dict)
|
71 |
|
72 |
-
return
|
73 |
|
74 |
embed_html = '<iframe src="https://www.youtube.com/embed/YOUTUBE_ID'\
|
75 |
'title="YouTube video player" frameborder="0" allow="accelerometer;'\
|
|
|
46 |
classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", top_k=None)
|
47 |
|
48 |
#Functions-----------------------------------------------------------------------------------------------------------------------
|
49 |
+
def transcribe(link,download,thumbnail,file,use_timestamps=True,sentiment_analysis=True):
|
50 |
'''inputs: file, return_timestamps'''
|
51 |
+
outputs = pipe(file, batch_size=BATCH_SIZE, generate_kwargs={"task": 'transcribe'}, return_timestamps=True)
|
52 |
text = outputs["text"]
|
53 |
timestamps = outputs["chunks"]
|
54 |
|
55 |
#If return timestamps is True, return html text with timestamps format
|
56 |
+
if use_timestamps==True:
|
57 |
spider_text = [f"{chunk['text']}" for chunk in timestamps] #Text for spider chart without timestamps
|
58 |
timestamps = [f"[{format_timestamp(chunk['timestamp'][0])} -> {format_timestamp(chunk['timestamp'][1])}] {chunk['text']}" for chunk in timestamps]
|
59 |
|
|
|
69 |
av_dict = calculate_average(trans_dict)
|
70 |
fig = spider_chart(av_dict)
|
71 |
|
72 |
+
return file, text, fig, av_dict
|
73 |
|
74 |
embed_html = '<iframe src="https://www.youtube.com/embed/YOUTUBE_ID'\
|
75 |
'title="YouTube video player" frameborder="0" allow="accelerometer;'\
|