Changed to load model from HF

README.md

@@ -1,5 +1,5 @@
 ---
-title: Image classification of four major lineages in the plum genus
+title: Prunus Lineage Classifier
 emoji: 🌸
 colorFrom: blue
 colorTo: gray

app.py

@@ -1,8 +1,9 @@
 import gradio as gr
 import numpy as np
 from fastai.learner import load_learner
+from huggingface_hub import from_pretrained_fastai
 
-learn_imp = load_learner('prunus_model_res50_temp_trop_v3_fp16_22epochs.pkl')
+learn_imp = from_pretrained_fastai('MikeTrizna/Prunus_lineage_classifier')
 
 def predict(image_np, description=None):
   classes = ['neotropical racemose', 'paleotropical racemose',