Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|