Update model v3: 10 classes

app.py

@@ -11,7 +11,7 @@ def handle_classify(image=None):
         image (gr.inputs.Image, optional): Input image to detect objects on. Defaults to None.
     """
 
-    model_path = "racist2.0.pt"
+    model_path = "racistv3.pt"
     model = YOLO(model_path)
 
     results = model(image)
@@ -23,7 +23,7 @@ def handle_classify(image=None):
     
     print(top5)
     
-    return "\n".join(["\t".join(row) for row in top5])
+    return "\n".join([f'{row[0]:<16}:{row[1]:>8.2f}' for row in top5])
 
 
 inputs = [
@@ -33,7 +33,7 @@ inputs = [
 
 outputs = gr.Textbox()
 
-title = "Racist model v2"
+title = "Racist model v3"
 
 SAMPLE_DIR = 'samples'
 examples = [os.path.join(SAMPLE_DIR, path) for path in os.listdir(SAMPLE_DIR)]

racist2.0.pt → racistv3.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6c59a98befcac1c90e9500f01516a88ae728eb3cea2a0e13c20d1f42d30efd41
-size 112503345
+oid sha256:a8d70d53a4961202f05e3fa697ea42a54bd1592a8770a27cfc0ad4addee3da12
+size 112509041