Vedmani commited on
Commit
8bbed7e
1 Parent(s): 6303db7

fixed flagging parameter

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def predict_image(array):
51
  input_image = Image.fromarray(array)
52
 
53
  # Load the model
54
- model = load_efficientnet_model('/home/vedmani/Downloads/efficientnet_epoch=18_loss=0.0020_val_f1score=0.8993.pth')
55
 
56
  # Transform the image
57
  transform = transforms.Compose([
 
51
  input_image = Image.fromarray(array)
52
 
53
  # Load the model
54
+ model = load_efficientnet_model('efficientnet_epoch=18_loss=0.0020_val_f1score=0.8993.pth')
55
 
56
  # Transform the image
57
  transform = transforms.Compose([