|
from flask import Flask,request,render_template,send_file,jsonify |
|
import os |
|
from transformers import AutoTokenizer, AutoModel |
|
|
|
app=Flask(__name__) |
|
MESSAGED={'title':'Script Server', |
|
'messageL':['script in "script" parameter or field of json input to /run will be run']} |
|
|
|
tokenizer = AutoTokenizer.from_pretrained('allenai/specter') |
|
encoder = AutoModel.from_pretrained('allenai/specter') |
|
|
|
@app.route('/encode',methods=['GET','POST']) |
|
def encode(): |
|
print(request) |
|
if request.method=='GET': |
|
text=request.args.get('text') |
|
elif request.method=='POST': |
|
data=request.get_json() |
|
if 'text' in data: text=data["text"] |
|
if text=='' or text is None: return -1 |
|
inputs = tokenizer(text, padding=True, truncation=True, |
|
return_tensors="pt", max_length=512) |
|
result = encoder(**inputs) |
|
embeddings = result.last_hidden_state[:, 0, :] |
|
emb_array = embeddings.detach().numpy() |
|
embedding=emb_array.tolist() |
|
return jsonify({'embedding': embedding}) |
|
|
|
@app.route("/file/<string:filename>") |
|
def return_file(filename): |
|
return send_file('./data/'+filename) |
|
|
|
@app.route('/run',methods=['GET','POST']) |
|
def run_script(): |
|
script='' |
|
|
|
print(request) |
|
if request.method=='GET': |
|
script=request.args.get('script') |
|
print('I am in get') |
|
elif request.method=='POST': |
|
print('I am in post') |
|
data=request.get_json() |
|
if 'script' in data: script=data['script'] |
|
if script=='' or script is None: return 'INVALID' |
|
os.system(script+' > ./out.txt') |
|
with open('./out.txt','r') as f: output=f.read() |
|
return output |
|
|
|
@app.route('/',methods=['GET', 'POST']) |
|
def home(): |
|
return render_template('home.html',messageD=MESSAGED) |
|
|
|
if __name__=='__main__': |
|
app.run(host="0.0.0.0", port=7860) |
|
|
|
|
|
|
