jpdiazpardo commited on
Commit
03978da
1 Parent(s): 258ccff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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(*args):
50
  '''inputs: file, return_timestamps'''
51
- outputs = pipe(args[3], 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 args[4]==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,7 +69,7 @@ def transcribe(*args):
69
  av_dict = calculate_average(trans_dict)
70
  fig = spider_chart(av_dict)
71
 
72
- return args[3], 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;'\
 
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;'\