adarshjha01 commited on
Commit
67b26c6
1 Parent(s): b0eb2d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import pipeline
8
 
9
  # Initialize summarization pipeline
10
  text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",
11
- torch_dtype=torch.bfloat16)
12
 
13
  def summarize_text(input_text):
14
  """Summarizes the given text."""
 
8
 
9
  # Initialize summarization pipeline
10
  text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",
11
+ torch_dtype=torch.bfloat16, device=-1)
12
 
13
  def summarize_text(input_text):
14
  """Summarizes the given text."""