Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, pipeline
|
|
7 |
MODEL_ID = "Xenova/distilbart-cnn-6-6"
|
8 |
# summariser = pipeline("summarization", model=MODEL_ID, device=-1) # -1 = CPU
|
9 |
tok = AutoTokenizer.from_pretrained(MODEL_ID)
|
10 |
-
model = ORTModelForSeq2SeqLM.from_pretrained(MODEL_ID
|
11 |
|
12 |
summariser = pipeline("summarization", model=model, tokenizer=tok, device=-1)
|
13 |
|
|
|
7 |
MODEL_ID = "Xenova/distilbart-cnn-6-6"
|
8 |
# summariser = pipeline("summarization", model=MODEL_ID, device=-1) # -1 = CPU
|
9 |
tok = AutoTokenizer.from_pretrained(MODEL_ID)
|
10 |
+
model = ORTModelForSeq2SeqLM.from_pretrained(MODEL_ID)
|
11 |
|
12 |
summariser = pipeline("summarization", model=model, tokenizer=tok, device=-1)
|
13 |
|