Spaces:
Running
Running
Update app.py
Browse files
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
|