Spaces:
Runtime error
Runtime error
Upload 3 files
Browse files- app.py +1 -1
- model_best.pth +3 -0
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from torchvision import transforms
|
|
| 7 |
from model import pixact
|
| 8 |
|
| 9 |
# Load the ONNX model
|
| 10 |
-
model_path = "./
|
| 11 |
|
| 12 |
transforms = transforms.Compose([
|
| 13 |
#transforms.Resize(size=(50, 50), antialias=True),
|
|
|
|
| 7 |
from model import pixact
|
| 8 |
|
| 9 |
# Load the ONNX model
|
| 10 |
+
model_path = "./model_best.pth" # Replace with your model path
|
| 11 |
|
| 12 |
transforms = transforms.Compose([
|
| 13 |
#transforms.Resize(size=(50, 50), antialias=True),
|
model_best.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d93877ca05698272600e8d80b42ee23479cc94a629c13926c1628896753dde8a
|
| 3 |
+
size 67952043
|