Spaces:

adarshjha01
/

Text-Summariser

Running

adarshjha01 commited on 3 days ago

Commit

8393be6

•

1 Parent(s): 2062dcb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import gradio as gr
 from transformers import pipeline
 text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",
-                torch_dtype = torch.bfloat16, device=0)
 def summary (input):
     output = text_summary(input)

 from transformers import pipeline
 text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",
+                torch_dtype = torch.bfloat16)
 def summary (input):
     output = text_summary(input)