Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ from PIL import Image, ImageEnhance, ImageFilter
|
|
| 11 |
from tqdm import tqdm
|
| 12 |
|
| 13 |
# Ensure TensorFlow runs on CPU
|
| 14 |
-
os.environ['CUDA_VISIBLE_DEVICES'] = '
|
| 15 |
|
| 16 |
class SwarmAgent:
|
| 17 |
def __init__(self, position, velocity):
|
|
|
|
| 11 |
from tqdm import tqdm
|
| 12 |
|
| 13 |
# Ensure TensorFlow runs on CPU
|
| 14 |
+
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
|
| 15 |
|
| 16 |
class SwarmAgent:
|
| 17 |
def __init__(self, position, velocity):
|