Spaces:
Runtime error
Runtime error
juan-op commited on
Commit ·
b8a736d
1
Parent(s): 6213c40
Specified that fp16 should not be used
Browse files
README.md
CHANGED
|
@@ -6,7 +6,7 @@ colorTo: gray
|
|
| 6 |
sdk: gradio
|
| 7 |
sdk_version: 3.16.0
|
| 8 |
app_file: app.py
|
| 9 |
-
pinned:
|
| 10 |
---
|
| 11 |
|
| 12 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
| 6 |
sdk: gradio
|
| 7 |
sdk_version: 3.16.0
|
| 8 |
app_file: app.py
|
| 9 |
+
pinned: true
|
| 10 |
---
|
| 11 |
|
| 12 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
app.py
CHANGED
|
@@ -23,7 +23,7 @@ def download_audio(url: str, path: str) -> None:
|
|
| 23 |
def transcribe(path: str) -> List[str]:
|
| 24 |
"""Transcribes the audio file at the given path and returns the text."""
|
| 25 |
model = whisper.load_model("tiny")
|
| 26 |
-
transcription = model.transcribe(path)["text"]
|
| 27 |
transcription_chunks = [transcription[i : i + 1000] for i in range(0, len(transcription), 1000)]
|
| 28 |
return transcription_chunks
|
| 29 |
|
|
@@ -42,7 +42,6 @@ def execute_pipeline(url: str) -> str:
|
|
| 42 |
download_audio(url, tmp_dir)
|
| 43 |
result = transcribe(f"{tmp_dir}/a.mp4")
|
| 44 |
text = summarize(result)
|
| 45 |
-
print("Done!")
|
| 46 |
return text
|
| 47 |
|
| 48 |
|
|
@@ -57,7 +56,6 @@ def main() -> None:
|
|
| 57 |
summarize_btn.click(fn=get_title, inputs=url, outputs=title)
|
| 58 |
summarize_btn.click(fn=execute_pipeline, inputs=url, outputs=output)
|
| 59 |
gr.Markdown("*Works best with videos under 10 minutes. It usually takes around 2-3 minutes to execute.*")
|
| 60 |
-
|
| 61 |
page.launch()
|
| 62 |
|
| 63 |
|
|
|
|
| 23 |
def transcribe(path: str) -> List[str]:
|
| 24 |
"""Transcribes the audio file at the given path and returns the text."""
|
| 25 |
model = whisper.load_model("tiny")
|
| 26 |
+
transcription = model.transcribe(path, fp16=False)["text"]
|
| 27 |
transcription_chunks = [transcription[i : i + 1000] for i in range(0, len(transcription), 1000)]
|
| 28 |
return transcription_chunks
|
| 29 |
|
|
|
|
| 42 |
download_audio(url, tmp_dir)
|
| 43 |
result = transcribe(f"{tmp_dir}/a.mp4")
|
| 44 |
text = summarize(result)
|
|
|
|
| 45 |
return text
|
| 46 |
|
| 47 |
|
|
|
|
| 56 |
summarize_btn.click(fn=get_title, inputs=url, outputs=title)
|
| 57 |
summarize_btn.click(fn=execute_pipeline, inputs=url, outputs=output)
|
| 58 |
gr.Markdown("*Works best with videos under 10 minutes. It usually takes around 2-3 minutes to execute.*")
|
|
|
|
| 59 |
page.launch()
|
| 60 |
|
| 61 |
|