ahmed-7124 commited on
Commit
ca891f7
·
verified ·
1 Parent(s): 7c565a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -13,8 +13,11 @@ classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnl
13
  image_model = timm.create_model('resnet50', pretrained=True)
14
  image_model.eval()
15
 
16
- # Load saved TensorFlow eye disease detection model (TensorFlow Model without Keras)
17
- eye_model = tf.keras.models.load_model('model')
 
 
 
18
 
19
  # Patient database
20
  patients_db = []
 
13
  image_model = timm.create_model('resnet50', pretrained=True)
14
  image_model.eval()
15
 
16
+ from tensorflow import keras
17
+ from tensorflow.keras.layers import TFSMLayer
18
+
19
+ # Load the model as a layer (in the SavedModel format)
20
+ eye_model = TFSMLayer(model_path='/content/saved_model')
21
 
22
  # Patient database
23
  patients_db = []