Made the image crop smaller

app.py

@@ -31,7 +31,7 @@ def make_prediction(img, feature_extractor, model):
 
 def fig2img(fig):
     buf = io.BytesIO()
-    fig.savefig(buf)
+    fig.savefig(buf, bbox_inches="tight")
     buf.seek(0)
     img = Image.open(buf)
     return img