Hexii commited on
Commit
241b950
1 Parent(s): f411139

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import os
9
 
10
 
11
  with open("classes.txt") as f:
12
- classes= [line.rstrip() for line in f]
13
 
14
  effnetb2, effnetb2_transforms = create_effnetb2_model(
15
  num_classes=len(classes))
 
9
 
10
 
11
  with open("classes.txt") as f:
12
+ classes= [line.rstrip() for line in f.readlines()]
13
 
14
  effnetb2, effnetb2_transforms = create_effnetb2_model(
15
  num_classes=len(classes))