Spaces:

KaburaJ
/

binary_image_classification_app

Runtime error

KaburaJ commited on Jan 5, 2023

Commit

3fcf295

1 Parent(s): 60fef1f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -111,7 +111,7 @@ def main():
             # predicted_idx = str(y_hat.item())
             # right_column.title("Prediction")
             # right_column.write(predicted_idx)
-            right_column.write(decode_predictions(outputs, top=3)[0])
 if __name__ == '__main__':

             # predicted_idx = str(y_hat.item())
             # right_column.title("Prediction")
             # right_column.write(predicted_idx)
+            right_column.write(decode_predictions(outputs, top=2)[0])
 if __name__ == '__main__':