Update server.js

server.js

@@ -3,7 +3,7 @@ import http from 'http';
 
 // Configuration
 const PORT = 7860;
-const MODEL_NAME = 'Xenova/codegen-2B-mono'; // ← Upgraded: 2B param model, much stronger code generation
+const MODEL_NAME = 'Xenova/starcoderbase-1b'; // ← 1B param, public, strong code generation
 let generator;
 
 // Initialize the model on startup
@@ -18,7 +18,7 @@ const server = http.createServer(async (req, res) => {
     res.setHeader('Access-Control-Allow-Origin', '*');
     res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
     res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
-    
+
     // Handle preflight requests for the API
     if (req.method === 'OPTIONS') {
         res.writeHead(200);
@@ -49,9 +49,9 @@ const server = http.createServer(async (req, res) => {
                     return res.end(JSON.stringify({ error: "Model is still loading..." }));
                 }
                 // Generate code based on the prompt
-                const output = await generator(prompt, { 
+                const output = await generator(prompt, {
                     max_new_tokens: 100,
-                    temperature: 0.7 
+                    temperature: 0.7
                 });
                 res.writeHead(200);
                 res.end(JSON.stringify({ result: output[0].generated_text }));