eesfeg commited on
Commit
a943095
·
1 Parent(s): 6952dc7

Add application file

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -11,17 +11,16 @@ 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
 
11
  import cv2
12
 
13
 
 
14
  from huggingface_hub import hf_hub_download
15
+ import tensorflow as tf
16
 
17
+ feature_extractor = hf_hub_download(
18
  repo_id="abdelac/FakeImageModelKeras",
19
  filename="feature_extractor.keras"
20
  )
21
 
22
+ feature_extractor = tf.keras.models.load_model(model_path)
23
+ print("Model loaded successfully!")
24
 
25
 
26
  # Load the models first model jjjff fkfkfkfkfkfk