micole66 commited on
Commit
31b427a
·
1 Parent(s): d18783f

Create new file

Browse files
Files changed (1) hide show
  1. app.py +14 -0
app.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import AutoTokenizer, AutoModelForCausalLM
3
+
4
+ model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-2.7B")
5
+ tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-2.7B")
6
+
7
+ def score(tokens_tensor):
8
+ loss=model(tokens_tensor, labels=tokens_tensor)[0]
9
+ return np.exp(loss.cpu().detach().numpy())
10
+
11
+ sentence = st.text_input("Enter the sentence","this is a well formed sentence.")
12
+ tokens_tensor = tokenizer.encode( sentence, add_special_tokens=False, return_tensors="pt")
13
+ res = score(tokens_tensor)
14
+ st.write('score is', res)