|
@ -1,5 +1,5 @@ |
|
|
import os |
|
|
import os |
|
|
os.environ["CUDA_VISIBLE_DEVICES"] = "0" |
|
|
os.environ["CUDA_VISIBLE_DEVICES"] = "3" |
|
|
from flask import Flask, jsonify |
|
|
from flask import Flask, jsonify |
|
|
from flask import request |
|
|
from flask import request |
|
|
from transformers import pipeline |
|
|
from transformers import pipeline |
|
|