Minbz commited on
Commit
e64bcbb
·
verified ·
1 Parent(s): 4a8a335

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ vit_b16 = hub.KerasLayer(model_url, trainable=False, input_shape=(224, 224, 3))
19
 
20
  # tf.config.experimental._could_use_gpu_kernel = False
21
 
22
- model = TFAutoModelForSequenceClassification.from_pretrained('model_pb_path')
23
 
24
  # model = tf.keras.models.load_model(model_keras_path, custom_objects={'KerasLayer': hub.KerasLayer})
25
  # model = tf.keras.models.load_model(model_h5_path)
 
19
 
20
  # tf.config.experimental._could_use_gpu_kernel = False
21
 
22
+ model = TFAutoModelForSequenceClassification.from_pretrained(model_pb_path)
23
 
24
  # model = tf.keras.models.load_model(model_keras_path, custom_objects={'KerasLayer': hub.KerasLayer})
25
  # model = tf.keras.models.load_model(model_h5_path)