nykadamec commited on
Commit
9fe9c9b
·
1 Parent(s): 6d8bed1

Update server.js

Browse files
Files changed (1) hide show
  1. server.js +34 -2
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
- // Pokud některé balíčky nefungují s ES import
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 {