Update app.py

app.py

@@ -8,11 +8,13 @@ from gradio.themes.monochrome import Monochrome
 from gradio.themes.default import Default
 from gradio.themes.utils import colors, fonts, sizes
 
+import spaces
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, AutoModelForTokenClassification, pipeline
 import os
 import colorsys
 import time
+
 def hex_to_rgb(hex_color: str) -> tuple[int, int, int]:
     hex_color = hex_color.lstrip('#')
     return tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))
@@ -107,6 +109,7 @@ tokenizer1 = AutoTokenizer.from_pretrained("AlGe/deberta-v3-large_Int_segment",
 
 model2 = AutoModelForSequenceClassification.from_pretrained("AlGe/deberta-v3-large_seq_ext", num_labels=1, token=auth_token)
 
+@spaces.GPU
 # Define functions to process inputs
 def process_ner(text, pipeline):
     output = pipeline(text)