Hugging Face's logo

Spaces:
zswwsz
/
Dissertation_img
Runtime error

App Files Community
zswwsz commited on
Commit
f38205a
•
1 Parent(s): 04ac1a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = 9
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"