Update app.py
Browse files
app.py
CHANGED
@@ -55,10 +55,10 @@ input_text = st.text_area('Please enter your text here')
|
|
55 |
preds = model(input_text)
|
56 |
|
57 |
# Select lab
|
58 |
-
def get_label(prediction_tensor):
|
59 |
-
|
60 |
#key = prediction_tensor.index(1)
|
61 |
#return group_dict[key]
|
62 |
|
63 |
-
|
64 |
-
st.text(get_label(preds))
|
|
|
55 |
preds = model(input_text)
|
56 |
|
57 |
# Select lab
|
58 |
+
#def get_label(prediction_tensor):
|
59 |
+
# print(prediction_tensor.index("1"))
|
60 |
#key = prediction_tensor.index(1)
|
61 |
#return group_dict[key]
|
62 |
|
63 |
+
st.text(preds)
|
64 |
+
#st.text(get_label(preds))
|