update app/meta to pull spaces model

README.md

@@ -8,6 +8,8 @@ sdk_version: 3.31.0
 app_file: app.py
 pinned: false
 python_version: 3.10
+models:
+  - joshvm/inaturalist_sgd_4k
 ---
 
 [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/metaformer-a-unified-meta-framework-for-fine/fine-grained-image-classification-on-cub-200)](https://paperswithcode.com/sota/fine-grained-image-classification-on-cub-200?p=metaformer-a-unified-meta-framework-for-fine)

app.py

@@ -2,12 +2,13 @@ from inference import Inference
 import argparse
 import gradio as gr
 import glob
+from huggingface_hub import hf_hub_download
 
 def parse_option():
     parser = argparse.ArgumentParser('MetaFG Inference script', add_help=False)
     parser.add_argument('--cfg', type=str, metavar="FILE", help='path to config file', default="configs/MetaFG_2_224.yaml")
     # easy config modification
-    parser.add_argument('--model-path', type=str, help="path to model data", default="./ckpt_4_mf2.pth")
+    parser.add_argument('--model-path', type=str, help="path to model data", default=None)
     parser.add_argument('--img-size', type=int, default=384, help='path to image')
     parser.add_argument('--meta-path', default="meta.txt", type=str, help='path to meta data')
     parser.add_argument('--names-path', default="names_mf2.txt", type=str, help='path to meta data')
@@ -17,19 +18,33 @@ def parse_option():
 if __name__ == '__main__':
     args = parse_option()
 
-    model = Inference(config_path=args.cfg,
-                       model_path=args.model_path,
-                       names_path=args.names_path)
+    if not args.model_path:
+        model_path = hf_hub_download(repo_id="joshvm/inaturalist_sgd_4k", filename="ckpt_epoch_35.pth")
+    else:
+        model_path = args.model_path
+
+    if not args.cfg:
+        model_config = hf_hub_download(repo_id="joshvm/inaturalist_sgd_4k", filename="MetaFG_2_224.yaml")
+    else:
+        model_config = args.cfg
+
+    if not args.names_path:
+        names_path = hf_hub_download(repo_id="joshvm/inaturalist_sgd_4k", filename="names_mf2.txt")
+    else:
+        names_path = args.names_path
+
+
+    model = Inference(config_path=model_config,
+                       model_path=model_path,
+                       names_path=names_path)
     
     def classify(image):
-        preds = model.infer(img_path=image, meta_data_path="meta.txt").squeeze()
-        print(len(model.classes))
-        print(model.classes)
-        confidences = {c: float(preds[i]) for i,c in enumerate(model.classes)}
+        preds = model.infer(img_path=image, meta_data_path="meta.txt")
+        #confidences = {c: float(preds[i]) for i,c in enumerate(model.classes)}
 
-        return confidences
+        return preds
     
-    gr.Interface(pfn=classify, 
+    gr.Interface(fn=classify, 
             inputs=gr.Image(shape=(args.img_size, args.img_size), type="pil"),
             outputs=gr.Label(num_top_classes=10),
             examples=glob.glob("./example_images/*")).launch()

meta.txt

@@ -0,0 +1 @@
+this is a picture of a plant or animal, taken in Singapore