Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from transformers import pipeline
|
|
8 |
# Use a pipeline as a high-level helper
|
9 |
from transformers import pipeline
|
10 |
|
11 |
-
|
12 |
|
13 |
# model_path=("../Models/models--sshleifer--distilbart-xsum-1-1/snapshots/891968fcbb0e421075cc2c3dfc8da8d4b24d54a4")
|
14 |
# text_summary = pipeline("summarization", model=model_path,
|
|
|
8 |
# Use a pipeline as a high-level helper
|
9 |
from transformers import pipeline
|
10 |
|
11 |
+
text_summary = pipeline("summarization", model="sshleifer/distilbart-xsum-1-1", torch_dtype=torch.bfloat16)
|
12 |
|
13 |
# model_path=("../Models/models--sshleifer--distilbart-xsum-1-1/snapshots/891968fcbb0e421075cc2c3dfc8da8d4b24d54a4")
|
14 |
# text_summary = pipeline("summarization", model=model_path,
|