Spaces:

griddava
/

pull-request-validator

Sleeping

Sgridda commited on Jul 11, 2025

Commit

2588618

1 Parent(s): 733f0e1

updated

Files changed (1) hide show

main.py CHANGED Viewed

@@ -37,6 +37,8 @@ def load_model():
     if model is None:
         print(f"Loading model: {MODEL_NAME} on device: {DEVICE}...")
         tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
         model = AutoModelForCausalLM.from_pretrained(
             MODEL_NAME,
             torch_dtype=torch.float32,
@@ -173,13 +175,12 @@ def root_html():
     </head>
     <body>
         <h1>AI Code Review Service</h1>
-        <p class="status">✅ Service is running in emergency mode</p>
         <h2>Available Endpoints:</h2>
         <div class="endpoint"><strong>GET /health</strong> - Health check</div>
         <div class="endpoint"><strong>POST /review</strong> - Submit code diff for review</div>
         <div class="endpoint"><strong>GET /docs</strong> - Interactive API documentation</div>
-        <div class="endpoint"><strong>GET /test</strong> - Simple test endpoint</div>
         <h2>Quick Test:</h2>
         <p><a href="/health">Test Health Endpoint</a></p>
@@ -187,9 +188,9 @@ def root_html():
         <h2>Status:</h2>
         <ul>
-            <li>Mode: Emergency (Mock responses)</li>
-            <li>AI Model: Disabled</li>
-            <li>Response Time: ~100ms</li>
         </ul>
     </body>
     </html>

     if model is None:
         print(f"Loading model: {MODEL_NAME} on device: {DEVICE}...")
         tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+        if tokenizer.pad_token is None:
+            tokenizer.pad_token = tokenizer.eos_token
         model = AutoModelForCausalLM.from_pretrained(
             MODEL_NAME,
             torch_dtype=torch.float32,
     </head>
     <body>
         <h1>AI Code Review Service</h1>
+        <p class="status">✅ Service is running with AI model</p>
         <h2>Available Endpoints:</h2>
         <div class="endpoint"><strong>GET /health</strong> - Health check</div>
         <div class="endpoint"><strong>POST /review</strong> - Submit code diff for review</div>
         <div class="endpoint"><strong>GET /docs</strong> - Interactive API documentation</div>
         <h2>Quick Test:</h2>
         <p><a href="/health">Test Health Endpoint</a></p>
         <h2>Status:</h2>
         <ul>
+            <li>Mode: AI-Powered</li>
+            <li>AI Model: Salesforce/codegen-350M-mono</li>
+            <li>Response Time: ~30-35 seconds</li>
         </ul>
     </body>
     </html>