Hugging Face's logo

Spaces:
darrenrourkeuk
/
xaidemo
Sleeping

App Files Community
darrenrourkeuk commited on
Commit
4291cab
·
verified ·
1 Parent(s): 4a7f1f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ model = AutoModelForImageClassification.from_pretrained(model_name).to(device)
32
  model.eval()
33
 
34
  # Print the model architecture for inspection.
35
- print(model)
36
 
37
  # Define custom preprocessing that matches the expected resolution (224x224) for ResNet50.
38
  custom_transform = transforms.Compose([
 
32
  model.eval()
33
 
34
  # Print the model architecture for inspection.
35
+ # print(model)
36
 
37
  # Define custom preprocessing that matches the expected resolution (224x224) for ResNet50.
38
  custom_transform = transforms.Compose([