Update app.py

app.py

@@ -9,11 +9,11 @@ import gradio as gr
 import spaces
 
 # Load the tokenizer and model
-led_tokenizer = AutoTokenizer.from_pretrained("allenai/led-base-16384-multi_lexsum-source-long")
+led_tokenizer = AutoTokenizer.from_pretrained("BAAI/bge-reranker-v2-m3")
 summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", tokenizer="sshleifer/distilbart-cnn-12-6", framework="pt")
 
 # Load the model separately
-model = AutoModelForSeq2SeqLM.from_pretrained("allenai/led-base-16384-multi_lexsum-source-long")
+model = AutoModelForSeq2SeqLM.from_pretrained("BAAI/bge-reranker-v2-m3")
 
 # Move the model to CUDA if available
 if torch.cuda.is_available():