Update app.py

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
 import torch
 import transformers
+import spaces
 from synthid_text import detector_bayesian, synthid_mixin, logits_processing
 
+
 # Configurations and model selection
 MODEL_NAME = "google/gemma-7b-it"  # Choose the model (Gemma models used in SynthID)
 DEVICE = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
@@ -21,6 +23,7 @@ logits_processor = logits_processing.SynthIDLogitsProcessor(
 )
 
 # Function to check for AI-generated content using SynthID
+@spaces.GPU
 def check_plagiarism(text):
     # Tokenize and process the input text
     tokens = tokenizer.encode_plus(text, return_tensors="pt", truncation=True, padding=True)