Add ZeroGPU decorator and space configuration

app.py

@@ -8,6 +8,8 @@ from datetime import datetime
 import json
 from typing import List, Dict
 import warnings
+from huggingface_hub.spaces import SpaceRuntime, SpaceStage
+from huggingface_hub.spaces.space_sdk import SpaceHardware
 
 # Filter out CUDA/NVML warnings
 warnings.filterwarnings('ignore', category=UserWarning)
@@ -29,6 +31,13 @@ os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:512"
 
+# Space configuration
+space_config = {
+    "hardware": SpaceHardware.A10G_SMALL,
+    "stage": SpaceStage.RUNTIME,
+    "runtime": SpaceRuntime.ZEROGPU
+}
+
 class Review:
     def __init__(self, code: str, language: str, suggestions: str):
         self.code = code
@@ -219,8 +228,12 @@ Code:
             'Device': str(self.device)
         }
 
-# Initialize reviewer
-reviewer = CodeReviewer()
+# Initialize reviewer with ZeroGPU
+@space_config
+def create_reviewer():
+    return CodeReviewer()
+
+reviewer = create_reviewer()
 
 # Create Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as iface: