trying to add bars in output

app.py

@@ -14,10 +14,9 @@ def classify_image(img):
  
  # Use the model to make a prediction
  prediction = model.predict(img)[0]
- #print(prediction)
  # Map the predicted class to a label
- dic = {'SFW': np.round(prediction[1],2), 'NSFW': np.round(prediction[0],2)}
- return dic#{'SFW': prediction[0][1], 'NSFW': prediction[0][0]}
+ dic = {'SFW': float(prediction[1]), 'NSFW': float(prediction[0])}
+ return dic
 
 def classify_url(url):
  # Load the image from the URL