Spaces:
Build error
Build error
import gradio as gr | |
import openai, os | |
from langchain.document_loaders.generic import GenericLoader | |
from langchain.document_loaders.parsers import OpenAIWhisperParser | |
from langchain.document_loaders.blob_loaders.youtube_audio import YoutubeAudioLoader | |
from dotenv import load_dotenv, find_dotenv | |
_ = load_dotenv(find_dotenv()) | |
openai.api_key = os.environ["OPENAI_API_KEY"] | |
def invoke(url): | |
url = url | |
save_dir = "docs/youtube/" | |
loader = GenericLoader( | |
YoutubeAudioLoader([url], save_dir), | |
OpenAIWhisperParser() | |
) | |
docs = loader.load() | |
return docs[0].page_content | |
description = """<a href='https://www.gradio.app/'>Gradio</a> UI using <a href='https://platform.openai.com/'>OpenAI</a> API with Whisper 1 foundation model.""" | |
gr.close_all() | |
demo = gr.Interface(fn=invoke, | |
inputs = [gr.Textbox(label = "YouTube URL", lines = 1)], | |
outputs = [gr.Textbox(label = "Automatic Speech Recognition", lines = 1)], | |
title = "Generative AI - Audio", | |
description = description) | |
demo.launch() |