Update app.py
Browse files
app.py
CHANGED
@@ -69,7 +69,7 @@ def sentence_prediction(sentence):
|
|
69 |
|
70 |
# outputs, [] = engine.predict_fn(test_data_loader, MODEL, device)
|
71 |
|
72 |
-
outputs =
|
73 |
|
74 |
print(outputs)
|
75 |
return outputs #{"label":outputs[0]}
|
|
|
69 |
|
70 |
# outputs, [] = engine.predict_fn(test_data_loader, MODEL, device)
|
71 |
|
72 |
+
outputs = classifier(sentence)
|
73 |
|
74 |
print(outputs)
|
75 |
return outputs #{"label":outputs[0]}
|