Spaces:

bstraehle
/

rag

Running

bstraehle commited on Oct 20, 2023

Commit

21a5617

•

1 Parent(s): a7466b5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,9 +12,9 @@ openai.api_key = os.environ["OPENAI_API_KEY"]
 def invoke(url):
     url = url
-    #save_dir = "docs/youtube/"
     loader = GenericLoader(
-        YoutubeAudioLoader([url]),
         OpenAIWhisperParser()
     )
     docs = loader.load()

 def invoke(url):
     url = url
+    save_dir = "docs/youtube/"
     loader = GenericLoader(
+        YoutubeAudioLoader([url], save_dir),
         OpenAIWhisperParser()
     )
     docs = loader.load()