Update app.py v4

app.py

@@ -29,12 +29,6 @@ def load_image_into_numpy_array(path):
     return pil_image_as_numpy_array(image)            
 
 def load_model():
-    download_dir = snapshot_download(REPO_ID)
-    saved_model_dir = os.path.join(download_dir, "saved_model")
-    detection_model = tf.saved_model.load(saved_model_dir)
-    return detection_model
-
-def load_model2():
     model_dir = 'saved_model'    
     detection_model = tf.saved_model.load(str(model_dir))
     return detection_model    
@@ -43,12 +37,7 @@ def load_model2():
 # image_path = 'test_samples/sample_balloon.jpeg
 # 
 
-def predict(pilimg):
-
-    image_np = pil_image_as_numpy_array(pilimg)
-    return predict2(image_np)
-
-def predict2(image_np):
+def predict(image_np):
 
     results = detection_model(image_np)
 
@@ -75,15 +64,14 @@ def predict2(image_np):
     return result_pil_img
 
 
-# REPO_ID = "amosfang/tfod_demo_hfspace"
-detection_model = load_model2()
+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')
 
-gr.Interface(fn=predict2,
+gr.Interface(fn=predict,
              inputs=gr.Image(type="pil"),
              outputs=gr.Image(type="pil")
              ).launch(share=True)