Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,7 +45,7 @@ def process(text):
|
|
| 45 |
"""
|
| 46 |
Translate incoming text to tokens and classify it
|
| 47 |
"""
|
| 48 |
-
pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
| 49 |
result = pipe(text)[0]
|
| 50 |
return format_predictions(top_pct(result))
|
| 51 |
|
|
|
|
| 45 |
"""
|
| 46 |
Translate incoming text to tokens and classify it
|
| 47 |
"""
|
| 48 |
+
pipe = pipeline("text-classification", model=model, tokenizer=tokenizer, top_k=None)
|
| 49 |
result = pipe(text)[0]
|
| 50 |
return format_predictions(top_pct(result))
|
| 51 |
|