Update app.py

app.py

@@ -1,4 +1,5 @@
 # This is a project of Chakra Lab LLC. All rights reserved.
+import spaces
 
 import gradio as gr
 import os
@@ -91,6 +92,7 @@ This policy is designed to determine whether or not content is hate speech.
 DEFAULT_CONTENT = "LLMs steal our jobs."
 
 # Function to make predictions
+@spaces.GPU
 def predict(content, policy):
     input_text = PROMPT.format(policy=policy, content=content)
     input_ids = tokenizer.encode(input_text, return_tensors="pt")