eesfeg commited on
Commit
aafeb7e
·
1 Parent(s): 824cd11

Add application file

Browse files
Files changed (1) hide show
  1. app.py +15 -1
app.py CHANGED
@@ -10,8 +10,22 @@ from tensorflow.keras.preprocessing.image import img_to_array
10
  import joblib
11
  import cv2
12
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  # Load the models first model jjjff fkfkfkfkfkfk
14
- feature_extractor = load_model("feature_extractor.keras", safe_mode=False)
15
  classifier = joblib.load("adaboost_model.pkl")
16
  label_encoder = joblib.load("label_encoder.pkl")
17
 
 
10
  import joblib
11
  import cv2
12
 
13
+
14
+
15
+ from huggingface_hub import hf_hub_download
16
+ from tensorflow.keras.models import load_model
17
+
18
+ model_path = hf_hub_download(
19
+ repo_id="abdelac/FakeImageModelKeras",
20
+ filename="feature_extractor.keras"
21
+ )
22
+
23
+ feature_extractor = load_model(model_path, safe_mode=False)
24
+
25
+
26
+
27
  # Load the models first model jjjff fkfkfkfkfkfk
28
+ # feature_extractor = load_model("feature_extractor.keras", safe_mode=False)
29
  classifier = joblib.load("adaboost_model.pkl")
30
  label_encoder = joblib.load("label_encoder.pkl")
31