Hugging Face's logo

Spaces:
jjuarez
/
test2
Runtime error

App Files Community
jjuarez commited on
Commit
bc8e7b8
•
1 Parent(s): e86bff1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import nltk
8
  nltk.download("punkt")
9
  raw_dataset = load_dataset("scientific_papers", "pubmed")
10
  metric = evaluate.load("rouge")
11
- model_checkpoint = "t5-base"
12
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
13
 
14
  if model_checkpoint in ["t5-small", "t5-base", "t5-large", "t5-3b", "t5-11b"]:
 
8
  nltk.download("punkt")
9
  raw_dataset = load_dataset("scientific_papers", "pubmed")
10
  metric = evaluate.load("rouge")
11
+ model_checkpoint = "t5-small"
12
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
13
 
14
  if model_checkpoint in ["t5-small", "t5-base", "t5-large", "t5-3b", "t5-11b"]: