add environment variables for cuda

app.py

@@ -30,6 +30,9 @@ from rocketshp.network import (
     calculate_centrality,
 )
 
+os.environ["OMP_NUM_THREADS"] = "4"
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"
+
 
 def plot_predictions(
     rmsf: np.ndarray,