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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ 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("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
9
+
10
 
11
  # Run locally
12
  # model_path = ("../Models/models--sshleifer--distilbart-cnn-12-6/snapshots/"