Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@ import gradio
|
|
| 3 |
from transformers import pipeline
|
| 4 |
|
| 5 |
# Used the model ID from the Hugging Face Hub
|
| 6 |
-
text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
|
| 7 |
|
| 8 |
#Example text
|
| 9 |
# text = '''Costello's (also known as Tim's) was a bar and restaurant in Midtown Manhattan,
|
|
|
|
| 3 |
from transformers import pipeline
|
| 4 |
|
| 5 |
# Used the model ID from the Hugging Face Hub
|
| 6 |
+
text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")#, torch_dtype=torch.bfloat16)
|
| 7 |
|
| 8 |
#Example text
|
| 9 |
# text = '''Costello's (also known as Tim's) was a bar and restaurant in Midtown Manhattan,
|