ldenoue commited on
Commit
52cd7a0
1 Parent(s): 2e0264f

flask does not work

Browse files
Files changed (1) hide show
  1. app.py +7 -15
app.py CHANGED
@@ -1,24 +1,16 @@
1
- #from fastapi import FastAPI
2
- from flask import Flask
3
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
4
  import numpy as np
5
 
6
- #app = FastAPI()
7
- app = Flask(__name__)
8
 
9
  model = GPT2LMHeadModel.from_pretrained('gpt2')
10
  tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
11
 
12
- @app.get("/score/<text>")
13
- def score(text):
14
- tokens_tensor = tokenizer.encode( text, add_special_tokens=False, return_tensors="pt")
15
  loss=model(tokens_tensor, labels=tokens_tensor)[0]
16
- print(text,loss)
17
- return str(np.exp(loss.cpu().detach().numpy()))
18
 
19
- #sentence = st.text_input("Enter the sentence","this is a well formed sentence.")
20
- #res = score(sentence)
21
- #st.write('score is', res)
22
-
23
- if __name__ == "__main__":
24
- app.run(debug=True)
 
1
+ import streamlit as st
 
2
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
3
  import numpy as np
4
 
 
 
5
 
6
  model = GPT2LMHeadModel.from_pretrained('gpt2')
7
  tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
8
 
9
+ def score(tokens_tensor):
 
 
10
  loss=model(tokens_tensor, labels=tokens_tensor)[0]
11
+ return np.exp(loss.cpu().detach().numpy())
 
12
 
13
+ sentence = st.text_input("Enter the sentence","this is a well formed sentence.")
14
+ tokens_tensor = tokenizer.encode( sentence, add_special_tokens=False, return_tensors="pt")
15
+ res = score(tokens_tensor)
16
+ st.write('score is', res)