yilunzhang
commited on
fixed depreciation
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ pipeline = pipeline(
|
|
17 |
|
18 |
def predict(text):
|
19 |
cleaned_text = clean_text(text)
|
20 |
-
predictions = pipeline(cleaned_text,
|
21 |
return {
|
22 |
p["label"]: p["score"] for p in predictions
|
23 |
}
|
|
|
17 |
|
18 |
def predict(text):
|
19 |
cleaned_text = clean_text(text)
|
20 |
+
predictions = pipeline(cleaned_text, top_k=None)[0]
|
21 |
return {
|
22 |
p["label"]: p["score"] for p in predictions
|
23 |
}
|