Update app.py

app.py

@@ -1,20 +1,11 @@
- 
 import gradio as gr
 import numpy as np
 import joblib
 import tensorflow as tf
-from huggingface_hub import hf_hub_download
 
-
-REPO_ID = "malware-USTH/mlp-malware" 
-
-# Download model & scaler from Hugging Face Hub
-model_dir = hf_hub_download(REPO_ID, "mlp_model.keras", repo_type="model")
-scaler_path = hf_hub_download(REPO_ID, "scaler.pkl")
-
-# Load
-model = tf.keras.models.load_model(model_dir)
-scaler = joblib.load(scaler_path)
+# Load local files 
+model = tf.keras.models.load_model("mlp_model.keras")
+scaler = joblib.load("scaler.pkl")
 
 N_FEATURES = model.input_shape[1]
 
@@ -33,5 +24,5 @@ inputs = [gr.Number(label=f"Feature {i+1}") for i in range(N_FEATURES)]
 gr.Interface(
     fn=predict,
     inputs=inputs,
-    outputs="json",
+    outputs="json"
 ).launch()