diff --git a/app.py b/app.py
index a1a156d..e085d4d 100644
--- a/app.py
+++ b/app.py
@@ -14,7 +14,7 @@ from multiprocessing import Process, Event
 
 app = Flask(__name__)
 
-device_count=0
+device_count = torch.cuda.device_count()
 workers = {}
 
 @app.route('/start/<id>')
@@ -191,7 +191,6 @@ def post_request(url, data):
 #             break
         
 if __name__ == '__main__':
-    device_count = torch.cuda.device_count()
     # cProfile.run('test_function()', 'test_function.profile')
     # p = pstats.Stats('test_function.profile')
     # p.sort_stats('cumulative').print_stats(100)  # Top 10 by cumulative time