Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 import numpy as np
 import tensorflow as tf
@@ -10,7 +11,7 @@ from PIL import Image, ImageEnhance, ImageFilter
 from tqdm import tqdm
 
 # Ensure TensorFlow runs on CPU
-tf.config.set_visible_devices([], 'GPU')
+os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
 
 class SwarmAgent:
     def __init__(self, position, velocity):