Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 import docx2txt
 
 tokenizer = AutoTokenizer.from_pretrained("allenai/led-base-16384")
-model = AutoModelForSeq2SeqLM.from_pretrained("checkpoint-64840").To("cpu")
+model = AutoModelForSeq2SeqLM.from_pretrained("checkpoint-64840").to("cpu")
 
 def summarize(text_file):
     file_extension = os.path.splitext(text_file.name)[1]
@@ -20,7 +20,7 @@ def summarize(text_file):
     else:
         raise ValueError(f"Unsupported file type: {file_extension}")
     
-    input_ids = tokenizer(text, return_tensors="pt").input_ids.to("cuda")
+    input_ids = tokenizer(text, return_tensors="pt").input_ids.to("cpu")
     global_attention_mask = torch.zeros_like(input_ids)
     # set global_attention_mask on first token
     global_attention_mask[:, 0] = 1