Update app.py

app.py

@@ -4,20 +4,18 @@ from huggingface_hub import from_pretrained_fastai
 
 learn = from_pretrained_fastai('mindwrapped/pokemon-card-checker')
 
-def check_card(input_img):
-    pred_label, _, score = learn.predict(input_img)
-    label_map = {
-      '1': 'real',
-      '0': 'fake'
-    }
-    return {'real': score[1], 'fake': score[0]}
+def check_card(image):
+    pred_label, _, scores = learn.predict(image)
+    scores = scores.detach().numpy()
+    return {'real': float(scores[1]), 'fake': float(scores[0])}
 
 demo = gr.Interface(
   fn=check_card, 
-  inputs=gr.Image(shape=(256, 256)),
+  inputs="image",
   outputs="label",
-  examples=['real-1.jpeg','real-2.jpeg','fake-1.jpeg','fake-2.jpeg'],
+  examples=['real-1.jpeg','real-2.jpeg','fake-1.jpeg','fake-2.jpeg','real-3.jpeg','real-4.jpeg','fake-3.jpeg','fake-4.jpeg'],
   title='Pokemon Card Checker',
-  description='A resnet34 model fine tuned to determine whether Pokemon cards are real or fake. Dataset is located [here](https://www.kaggle.com/datasets/ongshujian/real-and-fake-pokemon-cards).')
+  description='A resnet34 model fine-tuned to determine whether Pokemon cards are real or fake. Dataset is located [here](https://www.kaggle.com/datasets/ongshujian/real-and-fake-pokemon-cards).',
+  )
 
 demo.launch()