Update app.py

app.py

@@ -5,6 +5,7 @@ import requests
 import torch
 import gradio as gr
 import re
+import spaces
 
 model_id = "meta-llama/Llama-Guard-4-12B"
 
@@ -31,6 +32,7 @@ labels_dict = {
     "S13": "Elections",
 }
 
+@spaces.GPU
 def infer(image, text_input, model_output, exclude_categories):
     if image is None and text_input: 
         messages = [