Update app.py

app.py

@@ -1,12 +1,12 @@
-from transformers import LEDTokenizer, LEDForConditionalGeneration
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
 import re
 import gradio as gr
 import os
 import docx2txt
 
-tokenizer = LEDTokenizer.from_pretrained("nalmy/abstract-comp590")
-model = LEDForConditionalGeneration.from_pretrained("nalmy/abstract-comp590").to("cuda").half()
+tokenizer = AutoTokenizer.from_pretrained("allenai/led-base-16384")
+model = AutoModelForSeq2SeqLM.from_pretrained("checkpoint-64840").to("cuda").half()
 
 def summarize(text_file):
     file_extension = os.path.splitext(text_file.name)[1]