Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -19,6 +19,7 @@ with open("index_to_species.json", "r") as file:
|
|
19 |
index_to_species_data = file.read()
|
20 |
index_to_species = json.loads(index_to_species_data)
|
21 |
|
|
|
22 |
num_classes = len(list(index_to_species.keys()))
|
23 |
|
24 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
19 |
index_to_species_data = file.read()
|
20 |
index_to_species = json.loads(index_to_species_data)
|
21 |
|
22 |
+
print(index_to_species)
|
23 |
num_classes = len(list(index_to_species.keys()))
|
24 |
|
25 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|