kiranmadhusud commited on
Commit
1dcc426
·
1 Parent(s): 6237214

Fix: update RAG pipeline with TinyLlama text-generation

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -1,7 +1,16 @@
1
  import gradio as gr
2
  from pypdf import PdfReader
3
  from rag_pipeline import RAGPipeline
 
 
 
4
 
 
 
 
 
 
 
5
  rag = RAGPipeline()
6
 
7
  def process_file(file):
@@ -44,4 +53,4 @@ with gr.Blocks(title="📚 RAG App — No API Key") as demo:
44
  ask_btn.click(chat, inputs=[query_box, chatbot], outputs=chatbot)
45
  query_box.submit(chat, inputs=[query_box, chatbot], outputs=chatbot)
46
 
47
- demo.launch()
 
1
  import gradio as gr
2
  from pypdf import PdfReader
3
  from rag_pipeline import RAGPipeline
4
+ import warnings
5
+ import logging
6
+ import os
7
 
8
+ # Suppress FutureWarnings from torch/spaces
9
+ warnings.filterwarnings("ignore", category=FutureWarning)
10
+
11
+ # Suppress transformers verbose logs
12
+ logging.getLogger("transformers").setLevel(logging.ERROR)
13
+ os.environ["TOKENIZERS_PARALLELISM"] = "false"
14
  rag = RAGPipeline()
15
 
16
  def process_file(file):
 
53
  ask_btn.click(chat, inputs=[query_box, chatbot], outputs=chatbot)
54
  query_box.submit(chat, inputs=[query_box, chatbot], outputs=chatbot)
55
 
56
+ demo.launch(ssr_mode=False)