test

app.py

@@ -49,10 +49,8 @@ def predict(img, model_name):
         preds = anim_learn.predict(img)
 
     label_pred = str(preds[0])
-    acc = [float(preds[2][i]) for i in range(len(preds[2])) if preds[2][i] > 0.5]
-    pred = {label_pred: float(acc[0])}
         
-    return pred
+    return label_pred
 #########################################################################
 title = "TM-TKO Trademark Logo Image Classification Model"
 description = "Users can upload an image and corresponding image file name to get US design-code standard predictions on a trained model that utilizes the benchmark ResNet50 architecture."