kakil commited on
Commit
2a35400
·
verified ·
1 Parent(s): d6146bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  # Use a pipeline as a high-level helper
6
  from transformers import pipeline
7
 
8
- text_summary = pipline(task:"summarization", model=sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
9
 
10
  # Run locally
11
  # model_path = ("../Models/models--sshleifer--distilbart-cnn-12-6/snapshots/"
 
5
  # Use a pipeline as a high-level helper
6
  from transformers import pipeline
7
 
8
+ text_summary = pipline(task:"summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
9
 
10
  # Run locally
11
  # model_path = ("../Models/models--sshleifer--distilbart-cnn-12-6/snapshots/"