Hugging Face's logo

Spaces:
Shivam29rathore
/
doc-ai-10k_summariser
Runtime error

App Files Community
Shivam29rathore commited on
Commit
ac9c278
·
1 Parent(s): 1dbb181

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ checkpoint = "my_t5.sav"
6
 
7
  #load model from drive
8
  with open(checkpoint, "rb") as f:
9
- model= pickle.load(f, map_location='cpu')
10
 
11
  #tokenizer = AutoTokenizer.from_pretrained(checkpoint)
12
  #model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
 
6
 
7
  #load model from drive
8
  with open(checkpoint, "rb") as f:
9
+ model= pickle.load(f)
10
 
11
  #tokenizer = AutoTokenizer.from_pretrained(checkpoint)
12
  #model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)