David Li commited on
Commit
4865ee0
1 Parent(s): c98b0fc

fix: try again

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -92,6 +92,13 @@ def get_text_from_mp3_whisper(inputType:str, mp3_file: str, url_path: str, taskN
92
  # options = whisper.DecodingOptions(language="en", without_timestamps=True)
93
  options = dict(language=srcLanguage)
94
  transcribe_options = dict(task=taskName, **options)
 
 
 
 
 
 
 
95
  if inputType == "url":
96
  filename = get_all_files(url_path)
97
  result = model.transcribe(filename, **transcribe_options)
@@ -129,12 +136,14 @@ gr.Interface(
129
  title = 'Download Video From url and extract text from audio',
130
  fn=get_text_from_mp3_whisper,
131
  inputs=[
132
- gr.Dropdown(["url", "file"]),
133
  gr.inputs.Audio(type="filepath"),
134
  gr.inputs.Textbox(),
135
- gr.Dropdown(["translate", "transcribe"]),
136
- gr.Dropdown(["Japanese", "English"])
137
  ],
 
 
138
  outputs=[
139
  "json", "text", gr.outputs.Video(type="file")
140
  ],
 
92
  # options = whisper.DecodingOptions(language="en", without_timestamps=True)
93
  options = dict(language=srcLanguage)
94
  transcribe_options = dict(task=taskName, **options)
95
+ # return if url_path is not set, taskName is not set, srcLanguage is not set
96
+ if not url_path:
97
+ return "url_path is not set"
98
+ if not taskName:
99
+ return "taskName is not set"
100
+ if not srcLanguage:
101
+ return "srcLanguage is not set"
102
  if inputType == "url":
103
  filename = get_all_files(url_path)
104
  result = model.transcribe(filename, **transcribe_options)
 
136
  title = 'Download Video From url and extract text from audio',
137
  fn=get_text_from_mp3_whisper,
138
  inputs=[
139
+ gr.Dropdown(["url", "file"], value="url"),
140
  gr.inputs.Audio(type="filepath"),
141
  gr.inputs.Textbox(),
142
+ gr.Dropdown(["translate", "transcribe"], value="translate"),
143
+ gr.Dropdown(["Japanese", "English"], value="Japanese")
144
  ],
145
+ button_text="Go!",
146
+ button_color="#333333",
147
  outputs=[
148
  "json", "text", gr.outputs.Video(type="file")
149
  ],