Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -242,7 +242,7 @@ class mutil_model(nn.Module):
|
|
242 |
|
243 |
|
244 |
pth_path = './res2net_model_6_new.pt'
|
245 |
-
category_num =
|
246 |
|
247 |
# "cuda" only when GPUs are available.
|
248 |
#device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
242 |
|
243 |
|
244 |
pth_path = './res2net_model_6_new.pt'
|
245 |
+
category_num = 6
|
246 |
|
247 |
# "cuda" only when GPUs are available.
|
248 |
#device = "cuda" if torch.cuda.is_available() else "cpu"
|