Update app.py

app.py

@@ -1,9 +1,11 @@
-import torch
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-from optimum.bettertransformer import BetterTransformer
+import spaces
 import gradio as gr
 import pypdf
 import re
+import torch
+
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from optimum.bettertransformer import BetterTransformer
 
 tokenizer = AutoTokenizer.from_pretrained(
     "facebook/bart-large-cnn",
@@ -37,6 +39,7 @@ def extract_abstract(pdf_path):
     else:
         return ""
 
+@spaces.GPU
 def summarize_abstract(pdf_path):
     abstract_text = extract_abstract(pdf_path)