Update app.py

app.py

@@ -6,6 +6,7 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
 
+# Optimasi RAM
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 
@@ -21,12 +22,14 @@ app.add_middleware(
 class ItemRequest(BaseModel):
     items_text: str
 
+# Variable Global
 model = None
 encoder = None
 
 def get_model():
     global model, encoder
     if model is None:
+        # Load sangat normal, persis seperti di Colab
         model = tf.keras.models.load_model('model_strukly.keras')
         with open('penerjemah_kategori.pkl', 'rb') as file:
             encoder = pickle.load(file)
@@ -42,6 +45,7 @@ def classify_items(request: ItemRequest):
              
         m, e = get_model()
         
+        # Inference
         input_data = tf.constant([raw_items], dtype=tf.string)
         hasil_prediksi = m.predict(input_data, verbose=0)
         
@@ -51,10 +55,10 @@ def classify_items(request: ItemRequest):
         return {
             "success": True, 
             "class_id": category_id,
-            "class_name": category_name
+            "category_name": category_name
         }
         
-    except Exception as e:
+    except Exception as err:
         import traceback
         print(f"Error AI Klasifikasi: {traceback.format_exc()}")
-        return {"success": False, "error": str(e), "category_id": None, "category_name": None}
+        return {"success": False, "error": str(err), "category_id": None, "category_name": None}