Update app.py

app.py

@@ -4,7 +4,7 @@ import numpy as np
 from PIL import Image
 import io
 import sys
-from tensorflow.keras.applications.densenet import preprocess_input
+from tensorflow.keras.applications.resnet50 import preprocess_input
 
 # 1. Inisialisasi Aplikasi
 app = FastAPI(title="Ashoka Hipospadia Classifier API")
@@ -12,7 +12,7 @@ app = FastAPI(title="Ashoka Hipospadia Classifier API")
 # 2. Load Model
 print("Sedang memuat model...")
 try:
-    model = tf.keras.models.load_model('cnn_kfold_best_model_v2.h5')
+    model = tf.keras.models.load_model('cnn_kfold_best_model.h5')
     print("Model berhasil dimuat!")
 except Exception as e:
     print(f"Error memuat model: {e}")
@@ -103,8 +103,8 @@ async def predict(file: UploadFile = File(...)):
 def home():
     """Endpoint root untuk testing API"""
     return {
-        "message": "Ashoka Hipospadia Classifier API Online! 🚀\n",
-        "model": "DenseNet Binary Classification\n",
+        "message": "Ashoka Hipospadia Classifier API Online! 🚀",
+        "model": "ResNet50 Binary Classification",
         "classes": class_names
     }