Update app.py

app.py

@@ -102,6 +102,7 @@ class Predictor:
     def __init__(self):
         self.model_target_size = None
         self.last_loaded_repo = None
+        self.model = None  # Inisialisasi model di sini
 
     def download_model(self, model_repo):
         csv_path = huggingface_hub.hf_hub_download(
@@ -117,11 +118,11 @@ class Predictor:
         return csv_path, model_path
 
     def load_model(self, model_repo):
-        if model_repo == self.last_loaded_repo:
+         # Cek apakah model sudah dimuat
+        if model_repo == self.last_loaded_repo and self.model is not None:
             return
-
+        
         csv_path, model_path = self.download_model(model_repo)
-
         tags_df = pd.read_csv(csv_path)
         sep_tags = load_labels(tags_df)
 
@@ -130,12 +131,23 @@ class Predictor:
         self.general_indexes = sep_tags[2]
         self.character_indexes = sep_tags[3]
 
-        model = rt.InferenceSession(model_path)
-        _, height, width, _ = model.get_inputs()[0].shape
-        self.model_target_size = height
 
-        self.last_loaded_repo = model_repo
-        self.model = model
+        # Gunakan CPU execution provider jika GPU tidak tersedia
+        providers = ["CPUExecutionProvider"]
+        if rt.get_device() == "GPU":
+           providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
+        try:
+            model = rt.InferenceSession(model_path, providers=providers)
+            _, height, width, _ = model.get_inputs()[0].shape
+            self.model_target_size = height
+            self.last_loaded_repo = model_repo
+            self.model = model
+
+        except Exception as e:
+            print(f"Error loading model with given providers: {e}")
+            self.model = None
+            self.last_loaded_repo = None
+
 
     def prepare_image(self, image):
         target_size = self.model_target_size
@@ -179,6 +191,9 @@ class Predictor:
     ):
         self.load_model(model_repo)
 
+        if self.model is None:
+            return "", {}, {}, {}
+        
         image = self.prepare_image(image)
 
         input_name = self.model.get_inputs()[0].name
@@ -347,4 +362,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()