ohmygod0193 commited on
Commit
541dd53
·
1 Parent(s): ba125ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -262,11 +262,10 @@ def get_video(url):
262
  embed_html = '<iframe width="100%" height="315" src="https://www.youtube.com/embed/{}" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'.format(vid_id)
263
  return embed_html
264
 
265
- def summarize_youtube_video(url,force_transcribe,api_token="",
266
  temperature=0.2,words=300,do_sample=True):
267
  print("URL:",url)
268
- if api_token == "":
269
- api_token = HF_TOKEN
270
  title,text,transcript_source = transcribe_youtube_video(url,force_transcribe,True,api_token)
271
  print("Transcript:",text)
272
  summary, summary_source = summarize_text(title,text,temperature,words,True,api_token,do_sample)
@@ -351,7 +350,7 @@ with gr.Blocks() as demo:
351
 
352
  # Defining the functions to call on clicking the button
353
  sum_btn.click(fn=get_youtube_title, inputs=url, outputs=title, api_name="get_youtube_title", queue=False)
354
- sum_btn.click(fn=summarize_youtube_video, inputs=[url,force_transcribe,api_token,temperature,words,do_sample],
355
  outputs=[summary,transcript, transcript_source, summary_source], api_name="summarize_youtube_video", queue=True)
356
  sum_btn.click(fn=get_video, inputs=url, outputs=video, api_name="get_youtube_video", queue=False)
357
 
 
262
  embed_html = '<iframe width="100%" height="315" src="https://www.youtube.com/embed/{}" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'.format(vid_id)
263
  return embed_html
264
 
265
+ def summarize_youtube_video(url,force_transcribe,
266
  temperature=0.2,words=300,do_sample=True):
267
  print("URL:",url)
268
+ api_token = HF_TOKEN
 
269
  title,text,transcript_source = transcribe_youtube_video(url,force_transcribe,True,api_token)
270
  print("Transcript:",text)
271
  summary, summary_source = summarize_text(title,text,temperature,words,True,api_token,do_sample)
 
350
 
351
  # Defining the functions to call on clicking the button
352
  sum_btn.click(fn=get_youtube_title, inputs=url, outputs=title, api_name="get_youtube_title", queue=False)
353
+ sum_btn.click(fn=summarize_youtube_video, inputs=[url,force_transcribe,temperature,words,do_sample],
354
  outputs=[summary,transcript, transcript_source, summary_source], api_name="summarize_youtube_video", queue=True)
355
  sum_btn.click(fn=get_video, inputs=url, outputs=video, api_name="get_youtube_video", queue=False)
356