Spaces:

ravikagitha
/

Fake-Logo-Detection

Sleeping

ravikagitha commited on Mar 29, 2024

Commit

d4b20d3

verified ·

1 Parent(s): 23063bc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ def create_model():
     return clf_model
 clf_model = create_model()
-clf_model.load_weights('model.weights.h5')
 # input image preprocessing and prediction
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

     return clf_model
 clf_model = create_model()
+clf_model.load_weights('modelac90.weights.h5')
 # input image preprocessing and prediction
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')