Loc commited on
Commit
aeca5fd
·
1 Parent(s): b5e5457
Files changed (1) hide show
  1. app.py +3 -13
app.py CHANGED
@@ -1,15 +1,5 @@
1
  import gradio as gr
2
- from transformers import pipeline
 
3
 
4
- pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
5
-
6
- def predict(image):
7
- predictions = pipeline(image)
8
- return {p["label"]: p["score"] for p in predictions}
9
-
10
- gr.Interface(
11
- predict,
12
- inputs=gr.inputs.Image(label="Upload hot dog candidate", type="filepath"),
13
- outputs=gr.outputs.Label(num_top_classes=2),
14
- title="Hot Dog? Or Not?",
15
- ).launch()
 
1
  import gradio as gr
2
+ def sketch_recognition(img):
3
+ pass# Implement your sketch recognition model here...
4
 
5
+ gr.Interface(fn=sketch_recognition, inputs="sketchpad", outputs="label").launch()