ajdarshaydullin commited on
Commit
ad386f8
·
1 Parent(s): 15524fe

delete unused model

Browse files
Files changed (1) hide show
  1. app.py +0 -12
app.py CHANGED
@@ -5,18 +5,6 @@ from transformers import pipeline
5
 
6
  detector = pipeline("object-detection", model="hustvl/yolos-tiny")
7
 
8
- # Load model directly
9
- from transformers import AutoImageProcessor, AutoModelForImageClassification
10
-
11
- processor = AutoImageProcessor.from_pretrained("microsoft/resnet-50")
12
- model = AutoModelForImageClassification.from_pretrained("microsoft/resnet-50")
13
- def food_classifier(image):
14
- inputs = processor(image, return_tensors="pt")
15
- logits = model(**inputs).logits
16
- predicted_label = logits.argmax(-1).item()
17
- label = model.config.id2label[predicted_label]
18
- return [{'label': label, 'score': 0.0}]
19
-
20
 
21
  # Use a pipeline as a high-level helper
22
  from transformers import pipeline
 
5
 
6
  detector = pipeline("object-detection", model="hustvl/yolos-tiny")
7
 
 
 
 
 
 
 
 
 
 
 
 
 
8
 
9
  # Use a pipeline as a high-level helper
10
  from transformers import pipeline