yuhe6 commited on
Commit
50bf796
·
1 Parent(s): d3a747c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,8 +42,8 @@ def inference(input_image):
42
  probabilities = torch.nn.functional.softmax(output[0], dim=0)
43
 
44
  # Read the categories
45
- with open("dog_cat.txt", "r") as f:
46
- categories = [s.strip() for s in f.readlines()]
47
  with open("dog_cat.txt", "r") as f:
48
  categories = [s.strip() for s in f.readlines()]
49
  # Show top categories per image
 
42
  probabilities = torch.nn.functional.softmax(output[0], dim=0)
43
 
44
  # Read the categories
45
+ #with open("dog_cat.txt", "r") as f:
46
+ #categories = [s.strip() for s in f.readlines()]
47
  with open("dog_cat.txt", "r") as f:
48
  categories = [s.strip() for s in f.readlines()]
49
  # Show top categories per image