Spaces:

ShebMichel
/

classify_me

Sleeping

ShebMichel commited on Oct 10, 2024

Commit

15c7bdb

•

1 Parent(s): 97cbbcc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,6 +3,8 @@ from huggingface_hub import InferenceClient
 from transformers import pipeline
 def predict(image):
     predictions = pipeline(image)
     return {p["label"]: p["score"] for p in predictions}

 from transformers import pipeline
+pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 def predict(image):
     predictions = pipeline(image)
     return {p["label"]: p["score"] for p in predictions}