ashish-soni08 commited on
Commit
18e1eef
·
verified ·
1 Parent(s): bc6a573

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -5,7 +5,6 @@ from transformers import pipeline
5
  # Initialize Model
6
  get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", device=0)
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.
 
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.