add new beit model

app.py

@@ -120,6 +120,13 @@ def get_model(model_name):
                                   embedding_depth = 2,
                                   n_classes = n_classes)
         model.load_weights('model_classification/fossil-new.h5')
+    elif model_name == 'Fossils':
+        n_classes = 142
+        model = get_triplet_model_beit(input_shape = (384, 384, 3),
+                                  embedding_units = 256,
+                                  embedding_depth = 2,
+                                  n_classes = n_classes)
+        model.load_weights('model_classification/fossil-model.h5')
     else:
         raise ValueError(f"Model name '{model_name}' is not recognized") 
     return model,n_classes
@@ -151,6 +158,11 @@ def classify_image(input_image, model_name):
         model,n_classes = get_model(model_name)
         result = inference_resnet_finer_beit(input_image,model,size=384,n_classes=n_classes)
         return result
+    elif 'Fossils' ==model_name:
+        from inference_beit import inference_resnet_finer_beit
+        model,n_classes = get_model(model_name)
+        result = inference_resnet_finer_beit(input_image,model,size=384,n_classes=n_classes)
+        return result
     return None
 
 def get_embeddings(input_image,model_name):
@@ -174,6 +186,11 @@ def get_embeddings(input_image,model_name):
         model,n_classes = get_model(model_name)
         result = inference_resnet_embedding_beit(input_image,model,size=384,n_classes=n_classes)
         return result
+    elif 'Fossils' ==model_name:
+        from inference_beit import inference_resnet_embedding_beit
+        model,n_classes = get_model(model_name)
+        result = inference_resnet_embedding_beit(input_image,model,size=384,n_classes=n_classes)
+        return result
     return None
     
 
@@ -301,9 +318,9 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
                 
             with gr.Column():
                 model_name = gr.Dropdown(
-                    ["Mummified 170", "Rock 170","Fossils 142","Fossils new"],
+                    ["Mummified 170", "Rock 170","Fossils 142","Fossils new","Fossils"],
                     multiselect=False,
-                    value="Fossils new", # default option
+                    value="Fossils", # default option
                     label="Model",
                     interactive=True,
                     info="Choose the model you'd like to use"