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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  # app.py – CPU-only summariser for Hugging Face Spaces (free tier)
2
  from optimum.onnxruntime import ORTModelForSeq2SeqLM
3
  import textwrap, gradio as gr
4
- from transformers import pipeline
5
 
6
  # 1️⃣ small, open model that needs no access-token
7
  MODEL_ID = "Xenova/distilbart-cnn-6-6"
 
1
  # app.py – CPU-only summariser for Hugging Face Spaces (free tier)
2
  from optimum.onnxruntime import ORTModelForSeq2SeqLM
3
  import textwrap, gradio as gr
4
+ from transformers import AutoTokenizer, pipeline
5
 
6
  # 1️⃣ small, open model that needs no access-token
7
  MODEL_ID = "Xenova/distilbart-cnn-6-6"