Hugging Face's logo

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

App Files Community
Shivam29rathore commited on
Commit
23fe41c
·
1 Parent(s): ecc70cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,13 +2,13 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
 
3
 
4
 
5
- checkpoint = "Shivam29rathore/t5_10k_base.sav"
6
 
7
  #load model from drive
8
  #with open('Shivam29rathore/t5_10k_base', "rb") as f:
9
  #t5 = pickle.load(f)
10
 
11
- tokenizer = AutoTokenizer.from_pretrained(checkpoint)
12
  model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
13
 
14
 
 
2
 
3
 
4
 
5
+ checkpoint = "https://huggingface.co/Shivam29rathore/t5_10k_base"
6
 
7
  #load model from drive
8
  #with open('Shivam29rathore/t5_10k_base', "rb") as f:
9
  #t5 = pickle.load(f)
10
 
11
+ #tokenizer = AutoTokenizer.from_pretrained(checkpoint)
12
  model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
13
 
14