Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ labels = learner.dls.vocab
|
|
12 |
def predict(text):
|
13 |
output = learner.predict(text)
|
14 |
print("Output of learner.predict:", output)
|
15 |
-
|
16 |
pred, _, probs = output
|
17 |
elif len(output) == 2:
|
18 |
pred, probs = output
|
|
|
12 |
def predict(text):
|
13 |
output = learner.predict(text)
|
14 |
print("Output of learner.predict:", output)
|
15 |
+
if len(output) == 3:
|
16 |
pred, _, probs = output
|
17 |
elif len(output) == 2:
|
18 |
pred, probs = output
|