Browse Source

补充readme

master
majiahui@haimaqingfan.com 2 years ago
parent
commit
2a08f8bb36
  1. 2
      flask_batch.py

2
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

Loading…
Cancel
Save