Spaces:
Build error
Build error
yangxinsci1993
commited on
Commit
•
7df388d
1
Parent(s):
eafc169
Edit path
Browse files
app.py
CHANGED
|
@@ -4,8 +4,8 @@ from transformers import AutoModelForSeq2SeqLM
|
|
| 4 |
from transformers import AutoTokenizer
|
| 5 |
|
| 6 |
# Load trained model
|
| 7 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("
|
| 8 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 9 |
reframer = pipeline('summarization', model=model, tokenizer=tokenizer)
|
| 10 |
|
| 11 |
|
|
|
|
| 4 |
from transformers import AutoTokenizer
|
| 5 |
|
| 6 |
# Load trained model
|
| 7 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("output/reframer")
|
| 8 |
+
tokenizer = AutoTokenizer.from_pretrained("output/reframer")
|
| 9 |
reframer = pipeline('summarization', model=model, tokenizer=tokenizer)
|
| 10 |
|
| 11 |
|