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 |
|