update app.py

app.py

@@ -41,9 +41,9 @@ def load_model2():
     detection_model = tf.saved_model.load(str(model_dir))
     return detection_model    
 
-# samples_folder = 'test_samples
-# image_path = 'test_samples/sample_balloon.jpeg
-# 
+samples_folder = 'test_samples'
+image_path = 'test_samples/image489.png'
+
 
 def predict(pilimg):
 
@@ -79,11 +79,11 @@ def predict2(image_np):
 
 REPO_ID = "23b719w/assignment2_tfodmodel"
 detection_model = load_model()
-# pil_image = Image.open(image_path)
-# image_arr = pil_image_as_numpy_array(pil_image)
+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')
+predicted_img = predict(image_arr)
+predicted_img.save('predicted.jpg')
 
 gr.Interface(fn=predict,
              inputs=gr.Image(type="pil"),