Gregoryjr commited on
Commit
c8d475b
·
unverified ·
1 Parent(s): 45e0752

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ if con:
31
  model = AutoModelForSequenceClassification.from_pretrained("Greys/milestonemodel")
32
 
33
  def classify_sentence(text):
34
- inputs = tokenizer(sentence, return_tensors="pt")
35
  outputs = model(**inputs)
36
  probs = outputs.logits.softmax(dim=1)
37
  return probs.detach().numpy()[0]
 
31
  model = AutoModelForSequenceClassification.from_pretrained("Greys/milestonemodel")
32
 
33
  def classify_sentence(text):
34
+ inputs = tokenizer(text, return_tensors="pt")
35
  outputs = model(**inputs)
36
  probs = outputs.logits.softmax(dim=1)
37
  return probs.detach().numpy()[0]