sidd-harth011 commited on
Commit
4ca82bd
·
1 Parent(s): 4030d18
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -22,11 +22,11 @@ interpreter.allocate_tensors()
22
  input_details = interpreter.get_input_details()
23
  output_details = interpreter.get_output_details()
24
 
25
- # Download/load class indices
26
- class_indices_path = hf_hub_download(
27
- repo_id="sidd-harth011/checkingPDRMod",
28
- filename="class_indices.json"
29
- )
30
  class_indices = json.load(open(class_indices_path))
31
 
32
  # -----------------------------
 
22
  input_details = interpreter.get_input_details()
23
  output_details = interpreter.get_output_details()
24
 
25
+ # -----------------------------
26
+ # Load class indices locally from project repo
27
+ # -----------------------------
28
+ working_dir = os.path.dirname(os.path.abspath(__file__))
29
+ class_indices_path = os.path.join(working_dir, "class_indices.json")
30
  class_indices = json.load(open(class_indices_path))
31
 
32
  # -----------------------------