Spaces:

ilan541
/

OncUponTim

Sleeping

ilan541 commited on Jun 29, 2022

Commit

8de9764

1 Parent(s): 187aebc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,6 @@ from transformers import TFAutoModelForSequenceClassification
 model = TFAutoModelForSequenceClassification.from_pretrained("ilan541/OncUponTim")
 def predict(your_text):
-  df = split_text(text)
   inp = tokenizer(your_text, return_tensors='tf')
   if np.argmax(softmax(model(inp).logits)) == 0:
     return 'This content is not of high standard. It needs editing. '

 model = TFAutoModelForSequenceClassification.from_pretrained("ilan541/OncUponTim")
 def predict(your_text):
   inp = tokenizer(your_text, return_tensors='tf')
   if np.argmax(softmax(model(inp).logits)) == 0:
     return 'This content is not of high standard. It needs editing. '