Spaces:

mvaloatto
/

hotdog-or-not

Runtime error

mvaloatto commited on Jan 28, 2023

Commit

0f61155

1 Parent(s): 1dff3f9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,6 +3,8 @@ 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}

 pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
+with gr.Blocks(css=".gradio-container {background-color: red}") as demo
 def predict(image):
   predictions = pipeline(image)
   return {p["label"]: p["score"] for p in predictions}