Spaces:
Sleeping
Sleeping
Commit
·
ad386f8
1
Parent(s):
15524fe
delete unused model
Browse files
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
|