update example loading

Signed-off-by: peter szemraj <[email protected]>

utils.py

@@ -31,7 +31,7 @@ def truncate_word_count(text, max_words=512):
     return processed
 
 
-def load_examples(src):
+def load_examples(src, filetypes=[".txt", ".pdf"]):
     """
     load_examples - a helper function for the gradio module to load examples
     Returns:
@@ -39,14 +39,14 @@ def load_examples(src):
     """
     src = Path(src)
     src.mkdir(exist_ok=True)
-    examples = [f for f in src.glob("*.txt")]
+    examples = [f for f in src.iterdir() if f.suffix in filetypes]
     examples = natsorted(examples)
     # load the examples into a list
     text_examples = []
     for example in examples:
         with open(example, "r") as f:
             text = f.read()
-            text_examples.append([text, "large", 2, 512, 0.7, 3.5, 3])
+            text_examples.append([text, "base", 2, 1024, 0.7, 3.5, 3])
 
     return text_examples