Spaces:
Paused
Paused
Update server.js
Browse files
server.js
CHANGED
|
@@ -3,13 +3,15 @@ import http from 'http';
|
|
| 3 |
|
| 4 |
// Configuration
|
| 5 |
const PORT = 7860;
|
| 6 |
-
const MODEL_NAME = 'Xenova/
|
| 7 |
let generator;
|
| 8 |
|
| 9 |
// Initialize the model on startup
|
| 10 |
async function loadModel() {
|
| 11 |
console.log("Loading coding model...");
|
| 12 |
-
generator = await pipeline('text-generation', MODEL_NAME
|
|
|
|
|
|
|
| 13 |
console.log("Model loaded successfully!");
|
| 14 |
}
|
| 15 |
|
|
|
|
| 3 |
|
| 4 |
// Configuration
|
| 5 |
const PORT = 7860;
|
| 6 |
+
const MODEL_NAME = 'Xenova/codegen-350M-multi'; // ← Multi-language: JS, Python, C++, etc.
|
| 7 |
let generator;
|
| 8 |
|
| 9 |
// Initialize the model on startup
|
| 10 |
async function loadModel() {
|
| 11 |
console.log("Loading coding model...");
|
| 12 |
+
generator = await pipeline('text-generation', MODEL_NAME, {
|
| 13 |
+
model_file_name: 'decoder_model_merged' // ← Required for codegen models in Transformers.js v3
|
| 14 |
+
});
|
| 15 |
console.log("Model loaded successfully!");
|
| 16 |
}
|
| 17 |
|