add caching

gabrielchua · Sep 30, 2024 · 8bc56b2 · 8bc56b2
1 parent 410bed3
commit 8bc56b2
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 2 deletions.
diff --git a/app.py b/app.py
@@ -13,7 +13,6 @@
 # Third-party imports
 import gradio as gr
 from loguru import logger
-from openai import OpenAI
 from pydantic import BaseModel
 from pypdf import PdfReader
 from pydub import AudioSegment
@@ -185,7 +184,25 @@ def generate_podcast(
     allow_flagging="never",
     api_name="generate_podcast",
     theme=gr.themes.Soft(),
-    concurrency_limit=3
+    concurrency_limit=3,
+    examples=[
+        [
+            [str(Path("examples/1310.4546v1.pdf"))],
+            "",
+            "Fun",
+            "Short (1-2 min)",
+            "English"
+        ],
+        [
+            [],
+            "https://en.wikipedia.org/wiki/Hugging_Face",
+            "Fun",
+            "Short (1-2 min)",
+            "English"
+        ],
+    ],
+    cache_examples=True,
+    examples_cache_dir="examples_cached"
 )
 
 if __name__ == "__main__":

diff --git a/examples/1310.4546v1.pdf b/examples/1310.4546v1.pdf
diff --git a/examples_cached/.gitkeep b/examples_cached/.gitkeep