Simplify implementation and remove Space SDK dependencies

app.py

@@ -8,8 +8,6 @@ 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)
@@ -31,13 +29,6 @@ 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
@@ -228,12 +219,8 @@ Code:
             'Device': str(self.device)
         }
 
-# Initialize reviewer with ZeroGPU
-@space_config
-def create_reviewer():
-    return CodeReviewer()
-
-reviewer = create_reviewer()
+# Initialize reviewer
+reviewer = CodeReviewer()
 
 # Create Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as iface:

requirements.txt

@@ -12,7 +12,7 @@ einops>=0.7.0
 scipy>=1.11.0
 
 # Hugging Face
-huggingface-hub[cli,space]>=0.20.3
+huggingface-hub>=0.20.3
 
 # Utilities
 python-dotenv>=1.0.0