Update app.py

app.py

@@ -5,6 +5,10 @@ from PIL import Image
 from fastapi import FastAPI, UploadFile, File
 import io
 
+tf.config.experimental.set_memory_growth(
+    tf.config.list_physical_devices('CPU')[0], True
+)
+
 IMG_SIZE = 224
 
 # ===== LOAD MODEL =====