Hugging Face's logo

Spaces:
HimanshuGoyal2004
/
trashify_demo_v4
Running

App Files Community
HimanshuGoyal2004 commited on
Commit
45390d0
·
verified ·
1 Parent(s): e3b9430

Uploading Trashify box detection model app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ model_save_path = "HimanshuGoyal2004/rt_detrv2_finetuned_trashify_box_detector_v
15
 
16
  # Load the model and preprocessor
17
  # Because this app.py file is running directly on Hugging Face Spaces, the model will be loaded from the Hugging Face Hub
18
- image_processor = AutoImageProcessor.from_pretrained(model_save_path, force_download=true)
19
  model = AutoModelForObjectDetection.from_pretrained(model_save_path)
20
 
21
  # Set the target device (use CUDA/GPU if it is available)
 
15
 
16
  # Load the model and preprocessor
17
  # Because this app.py file is running directly on Hugging Face Spaces, the model will be loaded from the Hugging Face Hub
18
+ image_processor = AutoImageProcessor.from_pretrained(model_save_path)
19
  model = AutoModelForObjectDetection.from_pretrained(model_save_path)
20
 
21
  # Set the target device (use CUDA/GPU if it is available)