Update app.py

app.py

@@ -4,25 +4,22 @@ import numpy as np
 from ultralytics import YOLO
 import pandas as pd
 import pickle
-from huggingface_hub import hf_hub_download
 import random
 import torch.serialization
 
-# Download models from Hugging Face Hub
-model_repo_id = "your-username/fire-gas-detection-models"  # Replace with your repo ID
-rf_model_path = hf_hub_download(repo_id=model_repo_id, filename="model.pkl")
-yolo_model_path = hf_hub_download(repo_id=model_repo_id, filename="best.pt")
-
-# Load the Random Forest model
-with open(rf_model_path, "rb") as f:
-    rf_model = pickle.load(f)
+# Load the Random Forest model directly from the Space repository
+try:
+    with open("model.pkl", "rb") as f:
+        rf_model = pickle.load(f)
+except Exception as e:
+    raise Exception(f"Failed to load Random Forest model: {str(e)}")
 
-# Load the YOLO model with safe globals
+# Load the YOLO model with safe globals directly from the Space repository
 try:
     from ultralytics.nn.tasks import DetectionModel
     import torch.nn as nn
     torch.serialization.add_safe_globals([DetectionModel, nn.Sequential])
-    yolo_model = YOLO(yolo_model_path)
+    yolo_model = YOLO("best.pt")
 except Exception as e:
     raise Exception(f"Failed to load YOLO model: {str(e)}")
 
@@ -179,6 +176,6 @@ gr.Interface(
     fn=predict_system,
     inputs=inputs,
     outputs=outputs,
-    title "Fire & Gas Leak Detection System",
-    description "Simulate sensor data and upload an image for vision detection to predict the threat level using a Random Forest model and YOLOv10."
+    title="Fire & Gas Leak Detection System",
+    description="Simulate sensor data and upload an image for vision detection to predict the threat level using a Random Forest model and YOLOv10."
 ).launch()