Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-from keras.models import load_model
 from transformers import TFBertModel, TFXLMRobertaModel
 import numpy as np
 import tensorflow as tf
@@ -32,7 +31,7 @@ model_list = [
 ]
 
 def predict(text, target, chosen_model):    
-    model1 = load_model(chosen_model, custom_objects={"TFBertModel": TFBertModel})
+    model1 = tf.keras.models.load_model(chosen_model, custom_objects={"TFBertModel": TFBertModel})
     checkpoint = "neuralmind/bert-base-portuguese-cased"
     tokenizer = AutoTokenizer.from_pretrained(checkpoint, use_fast=True)
     tokpair = tokenizer(text, target, truncation=True, padding='max_length', return_tensors='np')