Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 import fitz
+import spaces
 
 # Initialize summarization pipeline
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
@@ -29,6 +30,7 @@ def summarize_pdf(pdf_file, model):
             text += page.get_text()
     return summarize_text(text, model)
 
+@spaces.GPU()
 def summarize(input_type, input_text, uploaded_file):
     try:
         if input_type == "Text":