Spaces:

edwardb1203
/

Project_Pollock

Sleeping

edwardb1203 commited on Apr 25, 2023

Commit

1832ead

1 Parent(s): a04f8da

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import os
 import docx2txt
 tokenizer = AutoTokenizer.from_pretrained("allenai/led-base-16384")
-model = AutoModelForSeq2SeqLM.from_pretrained("checkpoint-64840")
 def summarize(text_file):
     file_extension = os.path.splitext(text_file.name)[1]

 import docx2txt
 tokenizer = AutoTokenizer.from_pretrained("allenai/led-base-16384")
+model = AutoModelForSeq2SeqLM.from_pretrained("checkpoint-64840").To("cpu")
 def summarize(text_file):
     file_extension = os.path.splitext(text_file.name)[1]