Merge pull request #753 from axel7083/chore/adding-markdown-file-input

chore(summarizer): adding markdown type file input
containers · Aug 12, 2024 · 5671010 · 5671010
2 parents b61c908 + 4f0ecc8
commit 5671010
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recipes/natural_language_processing/summarizer/app/summarizer.py b/recipes/natural_language_processing/summarizer/app/summarizer.py
@@ -74,7 +74,7 @@ def read_file(file):
         pages = loader.load()
         text = "".join([p.page_content for p in pages]) 
 
-    if file_type == "text/plain":
+    if file_type in ["text/markdown", "text/plain"]:
         text = file.read().decode()   
 
     return text
@@ -86,7 +86,7 @@ def evaluate_summary(text, response):
     return score
 
 st.title("🔎 Summarizer")
-file = st.file_uploader("Upload file",type=[".txt",".pdf"])
+file = st.file_uploader("Upload file",type=[".txt",".pdf", ".md"])
 
 llm = ChatOpenAI(base_url=model_service,
              api_key="not required",