Spaces:
Sleeping
Sleeping
nykadamec
commited on
Commit
·
9fe9c9b
1
Parent(s):
6d8bed1
Update server.js
Browse files
server.js
CHANGED
|
@@ -10,8 +10,7 @@ import { createRequire } from 'module';
|
|
| 10 |
// Pro importy CommonJS balíčků
|
| 11 |
const require = createRequire(import.meta.url);
|
| 12 |
|
| 13 |
-
|
| 14 |
-
const somePackage = require('package-name');
|
| 15 |
|
| 16 |
// Import Upscaler using dynamic import
|
| 17 |
let Upscaler;
|
|
@@ -196,6 +195,39 @@ app.use((error, req, res, next) => {
|
|
| 196 |
res.status(500).json({ error: 'Internal server error' });
|
| 197 |
});
|
| 198 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 199 |
// Start server
|
| 200 |
async function startServer() {
|
| 201 |
try {
|
|
|
|
| 10 |
// Pro importy CommonJS balíčků
|
| 11 |
const require = createRequire(import.meta.url);
|
| 12 |
|
| 13 |
+
|
|
|
|
| 14 |
|
| 15 |
// Import Upscaler using dynamic import
|
| 16 |
let Upscaler;
|
|
|
|
| 195 |
res.status(500).json({ error: 'Internal server error' });
|
| 196 |
});
|
| 197 |
|
| 198 |
+
// Initialize TensorFlow.js
|
| 199 |
+
async function initializeTensorFlow() {
|
| 200 |
+
try {
|
| 201 |
+
console.log('Initializing TensorFlow.js...');
|
| 202 |
+
|
| 203 |
+
// Try WASM backend first
|
| 204 |
+
try {
|
| 205 |
+
await tf.setBackend('wasm');
|
| 206 |
+
await tf.ready();
|
| 207 |
+
console.log('TensorFlow.js initialized with WASM backend');
|
| 208 |
+
console.log('Current backend:', tf.getBackend());
|
| 209 |
+
return true;
|
| 210 |
+
} catch (wasmError) {
|
| 211 |
+
console.warn('WASM backend failed, trying CPU backend:', wasmError.message);
|
| 212 |
+
|
| 213 |
+
// Fallback to CPU backend
|
| 214 |
+
try {
|
| 215 |
+
await tf.setBackend('cpu');
|
| 216 |
+
await tf.ready();
|
| 217 |
+
console.log('TensorFlow.js initialized with CPU backend');
|
| 218 |
+
console.log('Current backend:', tf.getBackend());
|
| 219 |
+
return true;
|
| 220 |
+
} catch (cpuError) {
|
| 221 |
+
console.error('Both WASM and CPU backends failed:', cpuError.message);
|
| 222 |
+
return false;
|
| 223 |
+
}
|
| 224 |
+
}
|
| 225 |
+
} catch (error) {
|
| 226 |
+
console.error('Failed to initialize TensorFlow.js:', error);
|
| 227 |
+
return false;
|
| 228 |
+
}
|
| 229 |
+
}
|
| 230 |
+
|
| 231 |
// Start server
|
| 232 |
async function startServer() {
|
| 233 |
try {
|