Update app.py

app.py

@@ -102,8 +102,8 @@ with tab1:
             st.info("Please upload an image to classify.")
     else:
         st.info("Using Pretrained model")
-        # model = tf.keras.models.load_model("./models/medical_classifier.h5")
-        model = TFAutoModel.from_pretrained('./models/medical_classifier.h5')
+        model = tf.keras.models.load_model("./models/medical_classifier.h5")
+        # model = TFAutoModel.from_pretrained('./models/medical_classifier.h5')
         config = model.config
         if 'batch_shape' in config:
             config['input_shape'] = config.pop('batch_shape')[1:]  # Remove batch size