devdata commited on
Commit
5095ef5
·
1 Parent(s): 07fb3e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ feature_extractor = MobileNetV2FeatureExtractor.from_pretrained(model_folder)
11
  model = MobileNetV2ForImageClassification.from_pretrained(model_folder)
12
 
13
  # Load the image
14
- image = Image.open("/path/to/your/image.jpg")
15
 
16
  # Preprocess the image
17
  inputs = feature_extractor(images=image, return_tensors="pt")
 
11
  model = MobileNetV2ForImageClassification.from_pretrained(model_folder)
12
 
13
  # Load the image
14
+ image = Image.open("test_image.png")
15
 
16
  # Preprocess the image
17
  inputs = feature_extractor(images=image, return_tensors="pt")