Hot Dog or Not Classification Model (#1)

- feat: hot dog or not model classification (9415477dcf435e03f2345c264b8545db0bada5e6)
- feat: hot dog or not model classification (1fc6d29cdc99453c19d6ad7bef363547cc8080d4)

app.py

@@ -1,7 +1,21 @@
+# Hot Dog or Not Classification
+
 import gradio as gr
+from transformers import pipeline
+
+pipeline = pipeline(task="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 greet(name):
-	return "Hello from Hugging Face Spaces " + name + "!!"
+gradio_app = 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?',
+)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+if __name__ == '__main__':
+	gradio_app.launch()

hw.py

@@ -0,0 +1,7 @@
+import gradio as gr
+
+def greet(name):
+	return "Hello from Hugging Face Spaces " + name + "!!"
+
+iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+iface.launch()

requirements.txt

@@ -1 +1,3 @@
 gradio
+transformers
+torch