补充readme

2 years ago · 2a08f8bb36
1 changed files with 1 additions and 1 deletions
--- 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