Hugging Face's logo

Spaces:
binaychandra
/
draftnewapp
Sleeping

App Files Community
binaychandra commited on
Commit
7bdadbc
·
1 Parent(s): 32b7474

mobilenet update1

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from PIL import Image
5
  import requests
6
 
7
  # Load the pre-trained ResNet model
8
- model = models.MobileNet_V3_Small_Weights(pretrained=True)
9
  model.eval()
10
 
11
  # Define the transformation for input images
 
5
  import requests
6
 
7
  # Load the pre-trained ResNet model
8
+ model = models.mobilenet_v2(pretrained=True)
9
  model.eval()
10
 
11
  # Define the transformation for input images