Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from pathlib import Path
-from transformers import RAGTokenForingModel, AutoTokenizer, AutoModelForCausalLM
+from transformers import RagTokenForGeneration, AutoTokenizer, AutoModelForCausalLM
 from pdfplumber import open as open_pdf
 from typing import List
 
@@ -15,7 +15,7 @@ text_chunks: List[str] = [text[i:i+chunk_size] for i in range(0, len(text), chun
 
 # Load the RAG model and tokenizer for retrieval
 rag_tokenizer = AutoTokenizer.from_pretrained("facebook/rag-token-nq")
-rag_model = RAGTokenForingModel.from_pretrained("facebook/rag-token-nq")
+rag_model = RagTokenForGeneration.from_pretrained("facebook/rag-token-nq")
 
 # Load the DialoGPT model and tokenizer for generation
 dialogpt_tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")