Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 from loguru import logger
 from similarities import BertSimilarity, BM25Similarity
 
-from chatpdf import ChatPDF
+from chatpdf import Rag
 
 pwd_path = os.path.abspath(os.path.dirname(__file__))
 
@@ -31,7 +31,7 @@ if __name__ == '__main__':
     
     # Inicializar el modelo
     sim_model = BertSimilarity(model_name_or_path=args.sim_model_name, device=args.device)
-    model = ChatPDF(
+    model = Rag(
         similarity_model=sim_model,
         generate_model_type=args.gen_model_type,
         generate_model_name_or_path=args.gen_model_name,