anindya64 commited on
Commit
33d016a
1 Parent(s): 1732e5b

update app with new model.

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -9,13 +9,10 @@ description = """
9
 
10
  Simply upload an image of any person wearning/not-wearing helmet.
11
  """
 
12
 
13
- image_processor = AutoImageProcessor.from_pretrained(
14
- "devonho/detr-resnet-50_finetuned_cppe5"
15
- )
16
- model = AutoModelForObjectDetection.from_pretrained(
17
- "devonho/detr-resnet-50_finetuned_cppe5"
18
- )
19
 
20
  # Gradio Components
21
 
 
9
 
10
  Simply upload an image of any person wearning/not-wearing helmet.
11
  """
12
+ model_id = "anindya64/detr-resnet-50-finetuned"
13
 
14
+ image_processor = AutoImageProcessor.from_pretrained(model_id)
15
+ model = AutoModelForObjectDetection.from_pretrained(model_id)
 
 
 
 
16
 
17
  # Gradio Components
18