ashish-soni08 commited on
Commit
1b5cb5c
·
verified ·
1 Parent(s): c5bb2d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -3,9 +3,8 @@ import spaces
3
  from transformers import pipeline
4
 
5
  # Initialize Model
6
- get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", device=0, use_safetensors=True)
7
 
8
- @spaces.GPU(duration=120)
9
  def summarize(input: str) -> str:
10
  """
11
  Summarize the given input text using the sshleifer/distilbart-cnn-12-6 model.
 
3
  from transformers import pipeline
4
 
5
  # Initialize Model
6
+ get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", device=0)
7
 
 
8
  def summarize(input: str) -> str:
9
  """
10
  Summarize the given input text using the sshleifer/distilbart-cnn-12-6 model.