Update README.md
Browse files
    	
        README.md
    CHANGED
    
    | @@ -69,7 +69,7 @@ category_names = [ | |
| 69 | 
             
            prob_vector = probabilities[0].tolist()  # shape: (12,)
         | 
| 70 |  | 
| 71 | 
             
            predicted_labels = []
         | 
| 72 | 
            -
            for cat_name, prob | 
| 73 | 
             
                label = 1 if prob > threshold else 0
         | 
| 74 | 
             
                predicted_labels.append(label)
         | 
| 75 |  | 
|  | |
| 69 | 
             
            prob_vector = probabilities[0].tolist()  # shape: (12,)
         | 
| 70 |  | 
| 71 | 
             
            predicted_labels = []
         | 
| 72 | 
            +
            for cat_name, prob in zip(category_names, prob_vector):
         | 
| 73 | 
             
                label = 1 if prob > threshold else 0
         | 
| 74 | 
             
                predicted_labels.append(label)
         | 
| 75 |  | 

