test

app.py

@@ -3,15 +3,18 @@ from transformers import pipeline
 
 pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
-def predict(image):
+def predict(input_img):
     print('hello!')
-    print(image)
-    predictions = pipeline(image)
+    print(input_img)
+    predictions = pipeline(input_img)
     return {p["label"]: p["score"] for p in predictions}
 
-gr.Interface(
+gradio_app = gr.Interface(
     predict,
     inputs=gr.Image(label="Select hot dog candidate", type="pil"),
     outputs=gr.Label(num_top_classes=2),
     title="Hot Dog? Or Not?",
-).launch()
+)
+
+if __name__ == "__main__":
+    gradio_app.launch()