Update app.py

app.py

@@ -3,7 +3,8 @@ import gradio as gr
 from transformers import pipeline
 
 # Use a pipeline as a high-level helper
-text_summary = pipeline("summarization", model="Falconsai/text_summarization",torch_dtype=torch.bfloat16)
+device = 0 if torch.cuda.is_available() else -1
+text_summary = pipeline("summarization", model="Falconsai/text_summarization",device=device,torch_dtype=torch.bfloat16)
 
 def summary(input):
     output = text_summary(input)