Adding GPU support for hugging face

app.py

@@ -2,6 +2,14 @@ import gradio as gr
 from datetime import datetime
 from agent import tool_agent_loop, Email
 
+try:
+    import spaces
+except ImportError:
+    # Create a no-op decorator if spaces is not available
+    def spaces_gpu_decorator(func):
+        return func
+    spaces = type('spaces', (), {'GPU': spaces_gpu_decorator})()
+
 # System prompt that shows what tools the assistant has access to
 SYSTEM_PROMPT = """You are a helpful personal assistant with access to some email client tools which makes you a helpful personal email agent. The tools you have access to are:
 get_emails()
@@ -56,6 +64,7 @@ INBOX[2].timestamp = "2025-02-01 10:30:00"
 # This file now focuses on frontend components only
 # Backend logic has been moved to agent.py
 
+@spaces.GPU
 def submit_attack(attack_subject, attack_body, model_name="claude-3-5-sonnet-20241022"):
     """Process the attacker's email and return a response using the tool agent loop"""
     try:

instruction_classifier.py

@@ -22,6 +22,14 @@ try:
 except ImportError:
     hf_hub_download = None
 
+try:
+    import spaces
+except ImportError:
+    # Create a no-op decorator if spaces is not available
+    def spaces_gpu_decorator(func):
+        return func
+    spaces = type('spaces', (), {'GPU': spaces_gpu_decorator})()
+
 # Import required components from utils.py
 from utils import (
     TransformerInstructionClassifier, 
@@ -146,6 +154,7 @@ class InstructionClassifierSanitizer:
         self.model.to(self.device)
         self.model.eval()
     
+    @spaces.GPU
     def sanitize_tool_output(self, tool_output: str) -> str:
         """
         Main sanitization function that processes tool output and removes instruction content

requirements.txt

@@ -12,3 +12,4 @@ tqdm>=4.65.0
 datasets>=2.12.0
 accelerate>=0.20.0
 huggingface_hub>=0.20.0
+spaces