Update app.py

app.py

@@ -1,17 +1,15 @@
-import gradio as gr
 from transformers import pipeline
+import gradio as gr
+
 pipe = pipeline("image-classification", model="julien-c/hotdog-not-hotdog")
 
-def predict(input_img):
-  predictions = pipeline(input_img)
-  return input_img, {p["label"]:p["score"] for p in predictions}
+def predict(img):
+  predictions = pipe(img)
+  return img, {p["label"]: p["score"] for p in predictions}
 
-gradio_app = gr.Interface(
-     predict,
-     inputs = gr.Image(label="Select hot dog candidate", sources=['upload','webcam'],type="pil"),
-     outputs=[gr.Image(label="Processed Image"),gr.Label(label="Result",num_top_classes=2)],
-     title = "Hot Dog? or Not?"
-)
+interface = gr.Interface(fn = predict,
+                         inputs = gr.Image(label = "select hot dog image", sources = ['upload', 'webcam'], type='pil'),
+                         outputs = [gr.Image(label = "processed image"), gr.Label(label = "result", num_top_classes = 2)],
+                         title = "hot dog or not")
 
-if __name__ == "__main__":
-    gradio_app.launch()
+interface.launch()