Update app.py

app.py

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