Hugging Face's logo

Spaces:
edwardb1203
/
Project_Pollock
Running

App Files Community
1
edwardb1203 commited on
Commit
1832ead
•
1 Parent(s): a04f8da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import os
6
  import docx2txt
7
 
8
  tokenizer = AutoTokenizer.from_pretrained("allenai/led-base-16384")
9
- model = AutoModelForSeq2SeqLM.from_pretrained("checkpoint-64840")
10
 
11
  def summarize(text_file):
12
  file_extension = os.path.splitext(text_file.name)[1]
 
6
  import docx2txt
7
 
8
  tokenizer = AutoTokenizer.from_pretrained("allenai/led-base-16384")
9
+ model = AutoModelForSeq2SeqLM.from_pretrained("checkpoint-64840").To("cpu")
10
 
11
  def summarize(text_file):
12
  file_extension = os.path.splitext(text_file.name)[1]