Update app.py

app.py

@@ -36,9 +36,7 @@ def resizeImage(img):
     resized = cv2.resize(img, dim, interpolation = cv2.INTER_AREA)
     return resized
 
-from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
-tokenizer = AutoTokenizer.from_pretrained("shujianong/pkm-card")
-model = TFAutoModelForSequenceClassification.from_pretrained("shujianong/pkm-card")
+model = load_model("df_train.h5")
 
 """# Define function"""