diff --git a/flask_batch.py b/flask_batch.py
index 16f9e20..3708802 100644
--- a/flask_batch.py
+++ b/flask_batch.py
@@ -1,5 +1,5 @@
 import os
-os.environ["CUDA_VISIBLE_DEVICES"] = "0"
+os.environ["CUDA_VISIBLE_DEVICES"] = "3"
 from flask import Flask, jsonify
 from flask import request
 from transformers import pipeline