Hugging Face's logo

Spaces:
binaychandra
/
draftnewapp
Sleeping

App Files Community
binaychandra commited on
Commit
7442cbf
·
1 Parent(s): ae367d4

changed mobilenet

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.resnet50(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