adarshjha01
commited on
Commit
•
67b26c6
1
Parent(s):
b0eb2d1
Update app.py
Browse files
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."""
|