gregarific commited on
Commit
eb211b5
·
1 Parent(s): f74736b

Comment out load_model2()

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -34,12 +34,12 @@ def load_model():
34
  detection_model = tf.saved_model.load(saved_model_dir)
35
  return detection_model
36
 
37
- def load_model2():
38
- wget.download("https://nyp-aicourse.s3-ap-southeast-1.amazonaws.com/pretrained-models/balloon_model.tar.gz")
39
- tarfile.open("balloon_model.tar.gz").extractall()
40
- model_dir = 'saved_model'
41
- detection_model = tf.saved_model.load(str(model_dir))
42
- return detection_model
43
 
44
  # samples_folder = 'test_samples
45
  # image_path = 'test_samples/sample_balloon.jpeg
 
34
  detection_model = tf.saved_model.load(saved_model_dir)
35
  return detection_model
36
 
37
+ #def load_model2():
38
+ # wget.download("https://nyp-aicourse.s3-ap-southeast-1.amazonaws.com/pretrained-models/balloon_model.tar.gz")
39
+ # tarfile.open("balloon_model.tar.gz").extractall()
40
+ # model_dir = 'saved_model'
41
+ # detection_model = tf.saved_model.load(str(model_dir))
42
+ # return detection_model
43
 
44
  # samples_folder = 'test_samples
45
  # image_path = 'test_samples/sample_balloon.jpeg