Update main.py

main.py

@@ -10,7 +10,6 @@ import os
 from pathlib import Path
 from fastapi.middleware.cors import CORSMiddleware
 
-# Initialize FastAPI app (critical to name it 'app')
 app = FastAPI(title="Code Evaluation & Optimization API")
 
 # Required CORS configuration
@@ -35,12 +34,11 @@ try:
     tokenizer = AutoTokenizer.from_pretrained(
         MODEL_NAME,
         cache_dir=str(CACHE_DIR)
-    )
     model = AutoModelForCausalLM.from_pretrained(
         MODEL_NAME,
         device_map="auto",
         torch_dtype=torch.float16,
-        cache_dir=str(CACHE_DIR)
+        cache_dir=str(CACHE_DIR))
 except Exception as e:
     raise RuntimeError(f"Failed to load model: {str(e)}")
 
@@ -78,7 +76,6 @@ def evaluate_code(user_code: str, lang: str) -> dict:
             error_message = None if correctness else result.stderr.strip()
         else:
             return {"status": "error", "message": "Unsupported language", "score": 0}
-
     except Exception as e:
         return {"status": "error", "message": str(e), "score": 0}