Use only SSD MobileNet model

app.py

@@ -11,7 +11,7 @@ from huggingface_hub import snapshot_download
 import os
 import gradio as gr
 
-MODEL_REPO = 'sokonana/it107model'  # default model selected
+MODEL_REPO = 'sokonana/it107model'  
 PATH_TO_LABELS = 'data/label_map.pbtxt'   
 category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=True)
 
@@ -36,17 +36,12 @@ def load_model():
     return detection_model    
 
 
-def predict(pilimg, model):
-
-    if model == 'ResNet':
-        MODEL_REPO = 'sokonana/it107model2'
+def predict(pilimg):
 
     image_np = pil_image_as_numpy_array(pilimg)
     return predict2(image_np)
 
 def predict2(image_np):
-
-    detection_model = load_model()
     results = detection_model(image_np)
 
     # different object detection models have additional results
@@ -71,14 +66,9 @@ def predict2(image_np):
     
     return result_pil_img
 
-# detection_model = load_model()
-# pil_image = Image.open(image_path)
-# image_arr = pil_image_as_numpy_array(pil_image)
-
-# predicted_img = predict(image_arr)
-# predicted_img.save('predicted.jpg')
+detection_model = load_model()
 
 gr.Interface(fn=predict,
-             inputs=[gr.Image(type="pil"), gr.Radio(['MobileNet', 'ResNet'],label='Model Selection', value='MobileNet')],
+             inputs=[gr.Image(type="pil")],
              outputs=gr.Image(type="pil")
              ).launch(share=True)