Update main.py

main.py

@@ -7,18 +7,21 @@ import io
 import pandas as pd
 import os
 
+
 # Matiin log kalo run di Windows doang
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 
 app = FastAPI()
 
-model = load_model("foodnutritio_baru.h5")
 
-CLASS_NAMES = ['Ayam Bakar', 'Ayam Geprek', 'Ayam Goreng', 'Ayam Tepung', 'Bakso', 
-               'Chicken Katsu', 'Donat', 'Gado-Gado', 'Kopi', 'Mie Ayam', 'Mie Instan', 
-               'Nasi Goreng', 'Pecel Lele', 'Random Image', 'Rendang', 'Sate Ayam', 
-               'Sop', 'Soto', 'Telur Balado']
+model = load_model("foodnutrition.h5")
+
+CLASS_NAMES = ['Ayam Bakar', 'Ayam Geprek', 'Ayam Goreng', 'Ayam Tepung', 'Bakso', 'Bakwan',
+               'Batagor', 'Bolu Kukus', 'Burger', 'Cap Cay', 'Chicken Katsu', 'Cilok', 'Dimsum',
+               'Donat', 'Es Teh', 'Gado-Gado', 'Jasuke', 'Jus Jeruk', 'Kikil', 'Kopi', 'Mie Ayam', 'Mie Instan', 
+               'Nasi Goreng', 'Nasi Padang', 'Pecel Lele', 'Pizza', 'Random Image', 'Rendang', 'Risoles',
+               'Sate Ayam', 'Seblak', 'Siomay', 'Sop', 'Soto', 'Tahu','Telur Balado', 'Telur Dadar', 'Tempe']
 
 # Muat file CSV
 nutrisi_csv = pd.read_csv('nutrisi_origin.csv')
@@ -73,7 +76,7 @@ async def hello():
 async def predict(file: UploadFile = File(...)):
     img_bytes = await file.read()
     img = PIL.Image.open(io.BytesIO(img_bytes)).convert('RGB')
-    img = img.resize((128, 128))
+    img = img.resize((224, 224))
 
     img_array = np.array(img).astype("float32") / 255.0
     img_array = np.expand_dims(img_array, axis=0)
@@ -100,5 +103,4 @@ async def predict_text(nama_makanan: str = Form(...)):
         "prediction": nama_makanan,
         "confidence": 1.0,
         "nutrisi": data_nutrisi
-
     }