Spaces:
Sleeping
Sleeping
fix
Browse files- .gitignore +2 -1
- app.py +1 -1
.gitignore
CHANGED
|
@@ -2,4 +2,5 @@ __pycache__/
|
|
| 2 |
.DS_Store
|
| 3 |
.vscode/
|
| 4 |
|
| 5 |
-
test.*
|
|
|
|
|
|
| 2 |
.DS_Store
|
| 3 |
.vscode/
|
| 4 |
|
| 5 |
+
test.*
|
| 6 |
+
test/
|
app.py
CHANGED
|
@@ -16,7 +16,7 @@ def pokemon_classifier(inp):
|
|
| 16 |
labels = json.load(f)
|
| 17 |
|
| 18 |
x = TF.to_tensor(inp)
|
| 19 |
-
x = TF.resize(x, 64, antialias=True)
|
| 20 |
x = x.to(device)
|
| 21 |
x = x.unsqueeze(0)
|
| 22 |
|
|
|
|
| 16 |
labels = json.load(f)
|
| 17 |
|
| 18 |
x = TF.to_tensor(inp)
|
| 19 |
+
x = TF.resize(x, (64, 64), antialias=True)
|
| 20 |
x = x.to(device)
|
| 21 |
x = x.unsqueeze(0)
|
| 22 |
|