Spaces:

SondosM
/

API_2x1

Sleeping

SondosM commited on Apr 20

Commit

d3c7920

verified ·

1 Parent(s): 7871561

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -91,9 +91,9 @@ WILOR_CFG         = get_hf_file("pretrained_models/pretrained_models/model_confi
 DETECTOR_PATH     = get_hf_file("pretrained_models/pretrained_models/detector.pt")
 # Classifiers
-CLASSIFIER_PATH   = get_hf_file("classifier.pkl")
-MLP_LETTERS_PATH  = get_hf_file("MLP_letters.pkl")
-MLP_NUMBERS_PATH  = get_hf_file("MLP_numbers.pkl")
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

 DETECTOR_PATH     = get_hf_file("pretrained_models/pretrained_models/detector.pt")
 # Classifiers
+MLP_LETTERS_PATH  = "MLP_letters.pkl"
+MLP_NUMBERS_PATH  = "MLP_numbers.pkl"
+CLASSIFIER_PATH   = "classifier.pkl"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"