Update app.py
Browse files
app.py
CHANGED
|
@@ -56,7 +56,7 @@ 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 |
|
|
|
|
| 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 |
|