Aneeshmishra commited on
Commit
cb73ab4
·
verified ·
1 Parent(s): c969a0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, file_name="model.onnx")
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