Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch.nn.functional as F
@@ -13,7 +14,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 model.eval()
 
-
+@spaces.GPU
 def predict(title, abstract):
     text = f'''Given a certain paper, Title: {title}\n Abstract: {abstract}. \n Predict its normalized academic impact (between 0 and 1):'''
     inputs = tokenizer(text, return_tensors="pt")