EkhiAzur commited on
Commit
92cbc2e
·
1 Parent(s): 9b7b0c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ classifier = pipeline("text-classification", model=model, tokenizer=tokenizer, m
16
  padding=True, truncation=True, batch_size=1)
17
 
18
  def prozesatu(testua):
19
- prediction = prozesatu.classifier(testua)
20
  return f'C1:{prediction["label"]}. Probabilitatea:{prediction["score"]}'
21
 
22
  prozesatu.classifier = classifier
 
16
  padding=True, truncation=True, batch_size=1)
17
 
18
  def prozesatu(testua):
19
+ prediction = prozesatu.classifier([testua])
20
  return f'C1:{prediction["label"]}. Probabilitatea:{prediction["score"]}'
21
 
22
  prozesatu.classifier = classifier