AngelinaZanardi commited on
Commit
5f56b9e
·
verified ·
1 Parent(s): 87ec1bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -19,6 +19,8 @@ import yt_dlp # Added import for yt-dlp
19
  MODEL_NAME = "NbAiLab/nb-whisper-large"
20
  lang = "no"
21
 
 
 
22
  share = (os.environ.get("SHARE", "False")[0].lower() in "ty1") or None
23
  auth_token = os.environ.get("AUTH_TOKEN") or True
24
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
@@ -62,7 +64,7 @@ def transcribe(file, return_timestamps=False):
62
  line = f"[{start_time} -> {end_time}] {chunk['text']}"
63
  text.append(line)
64
  formatted_text = "\n".join(text)
65
- formatted_text += "\n\n<i>Transkribert med NB-Whisper demo</i>"
66
  return formatted_text
67
 
68
  def _return_yt_html_embed(yt_url):
@@ -98,14 +100,14 @@ def yt_transcribe(yt_url, return_timestamps=False):
98
  demo = gr.Blocks()
99
 
100
  with demo:
101
- gr.HTML('<center><img src="Logo.png" alt="Nasjonalbiblioteket Logo" width="300"></center>')
102
  mf_transcribe = gr.Interface(
103
  fn=transcribe,
104
  inputs=[
105
  gr.components.Audio(sources=['upload', 'microphone'], type="filepath"),
106
  gr.components.Checkbox(label="Inkluder tidsstempler"),
107
  ],
108
- outputs="text",
109
  title="NB-Whisper",
110
  description=(
111
  "Transkriber lange lydopptak fra mikrofon eller lydfiler med et enkelt klikk! Demoen bruker den fintunede"
 
19
  MODEL_NAME = "NbAiLab/nb-whisper-large"
20
  lang = "no"
21
 
22
+ logo_path = os.path.join(os.path.dirname(__file__), "Logo.png")
23
+
24
  share = (os.environ.get("SHARE", "False")[0].lower() in "ty1") or None
25
  auth_token = os.environ.get("AUTH_TOKEN") or True
26
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
64
  line = f"[{start_time} -> {end_time}] {chunk['text']}"
65
  text.append(line)
66
  formatted_text = "\n".join(text)
67
+ formatted_text += "<br><br><i>Transkribert med NB-Whisper demo</i>"
68
  return formatted_text
69
 
70
  def _return_yt_html_embed(yt_url):
 
100
  demo = gr.Blocks()
101
 
102
  with demo:
103
+ gr.Image(value=logo_path, type="filepath", elem_id="logo", label=None).style(width=100) # Adjust width as needed
104
  mf_transcribe = gr.Interface(
105
  fn=transcribe,
106
  inputs=[
107
  gr.components.Audio(sources=['upload', 'microphone'], type="filepath"),
108
  gr.components.Checkbox(label="Inkluder tidsstempler"),
109
  ],
110
+ outputs=gr.HTML(label="text"),
111
  title="NB-Whisper",
112
  description=(
113
  "Transkriber lange lydopptak fra mikrofon eller lydfiler med et enkelt klikk! Demoen bruker den fintunede"