Bitirme commited on
Commit
f48495a
1 Parent(s): 5fefbf7

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +2 -2
api.py CHANGED
@@ -46,13 +46,13 @@ def result(predictions):
46
 
47
  # Model tanımlamaları
48
  models_names = ["ODIR-B-2K-5Class-LastTrain-Xception", "ODIR-B-2K-6Class-LastTrain-Xception"]
49
- model_paths = ["ODIR-B-2K-5Class-LastTrain-Xception", "ODIR-B-2K-6Class-LastTrain-Xception"]
50
 
51
  filters = [filter1, filter2] # İhtiyacınıza göre filtre fonksiyonları burada tanımlandı
52
 
53
  models = []
54
  for model_path in model_paths:
55
- model = tf.saved_model.load(model_path)
56
  models.append(model)
57
 
58
  class PredictionResponse(BaseModel):
 
46
 
47
  # Model tanımlamaları
48
  models_names = ["ODIR-B-2K-5Class-LastTrain-Xception", "ODIR-B-2K-6Class-LastTrain-Xception"]
49
+ model_paths = ["Bitirme/odirmodel/ODIR-B-2K-5Class-LastTrain-Xception", "Bitirme/odirmodel/ODIR-B-2K-6Class-LastTrain-Xception"]
50
 
51
  filters = [filter1, filter2] # İhtiyacınıza göre filtre fonksiyonları burada tanımlandı
52
 
53
  models = []
54
  for model_path in model_paths:
55
+ model = from_pretrained_keras(model_path)
56
  models.append(model)
57
 
58
  class PredictionResponse(BaseModel):