Raiff1982 commited on
Commit
21e75a0
·
verified ·
1 Parent(s): ec7abe6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -17,11 +17,6 @@ for gpu in gpus:
17
  tf.config.experimental.set_memory_growth(gpu, True)
18
  except RuntimeError as e:
19
  print(f"[TF] GPU memory growth config error: {e}")
20
-
21
- # Limit CPU memory usage (for Hugging Face Spaces)
22
- tf.config.set_logical_device_configuration(
23
- tf.config.list_physical_devices('CPU')[0],
24
- [tf.config.LogicalDeviceConfiguration(memory_limit=2048)]
25
  )
26
  # Initialize AI Core for TB analysis
27
  ai_core = AICoreAGIX()
 
17
  tf.config.experimental.set_memory_growth(gpu, True)
18
  except RuntimeError as e:
19
  print(f"[TF] GPU memory growth config error: {e}")
 
 
 
 
 
20
  )
21
  # Initialize AI Core for TB analysis
22
  ai_core = AICoreAGIX()