Spaces:
Running
Running
Update app.py
Browse files
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(
|
| 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)
|