= commited on
Commit
87265a8
1 Parent(s): 8b7c7b2

fixed model name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from timeit import default_timer as timer
13
 
14
  class_names = ['Honda', 'Hyundai', 'Toyota']
15
 
16
- effnet_v2, transforms = create_effnet_v2_model(num_classes=len(class_names), weights_path="efficient_net_v2_s_carvision_trivialaug_label_smoothing.pth")
17
 
18
  def predict(image):
19
 
 
13
 
14
  class_names = ['Honda', 'Hyundai', 'Toyota']
15
 
16
+ effnet_v2, transforms = create_effnet_v2_model(num_classes=len(class_names), weights_path="efficient_net_s_carvision_3.pth")
17
 
18
  def predict(image):
19