Upload app.py

app.py

@@ -1,10 +1,10 @@
 import os
 import gradio as gr
 from transformers.pipelines import pipeline
-from langchain.document_loaders import PyPDFLoader
+from langchain_community.document_loaders import PyPDFLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.vectorstores import FAISS
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_community.vectorstores import FAISS
 from langchain.chains import RetrievalQA
 from langchain_community.llms import HuggingFacePipeline
 
@@ -51,7 +51,7 @@ C. General statistics
 Question: "{question}"
 Answer with only A, B, or C.
 """
-    result = llm_pipeline(routing_prompt, max_length=10, do_sample=False)
+    result = llm_pipeline(routing_prompt, max_new_tokens=30, do_sample=False)
     route = result[0]['generated_text'].strip().upper()
     if route.startswith("A"):
         return "stat6371"