Update app.py

app.py

@@ -37,7 +37,7 @@ def load_image_into_numpy_array(path):
     return pil_image_as_numpy_array(image)            
 
 def load_model():
-    download_dir = snapshot_download()
+    download_dir = snapshot_download(REPO_ID)
     saved_model_dir = os.path.join(download_dir, "saved_model.pb")
     detection_model = tf.saved_model.load(saved_model_dir)
     return detection_model
@@ -54,13 +54,12 @@ detection_model = load_model()
 # predicted_img = predict(image_arr)
 # predicted_img.save('predicted.jpg')
 
-
+REPO_ID = "23A066X"
 gr.Interface(fn=predict,
              inputs=[gr.Image(type="pil",label="Input Image"),gr.Textbox(placeholder="0.50",label="Set the confidence threshold (0.00-1.00)")],
              outputs=gr.Image(type="pil",label="Output Image"), 
              title="Cauliflower and Beetroot Detection",
              description="Model: ssd_resnet50_v1_fpn_640x640_coco17_tpu-8",
-             theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
-             examples=[["test_samples/image489.png",0.55], ["test_samples/image825.png",0.55], ["test_samples/image833.png",0.55], ["test_samples/image846.png",0.55]]
+             theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky")
              ).launch(share=True)