sraimund commited on
Commit
f4e7b13
·
verified ·
1 Parent(s): 2989568

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -24,5 +24,10 @@ def predict(input_image):
24
 
25
  return "Map" if result == 1 else "No map"
26
 
27
- demo = gr.Interface(fn=predict, inputs=gr.Image(title="Map identification", description="The model predicts whether an image is a map or not. It takes about 30 seconds since it runs on a CPU (it is much faster on a GPU).", article="<a href='https://huggingface.co/datasets/sraimund/MapPool'>MapPool - Bubbling up an extremely large corpus of maps for AI</a>", label="Input image", type="pil"), outputs="text")
 
 
 
 
 
28
  demo.launch()
 
24
 
25
  return "Map" if result == 1 else "No map"
26
 
27
+ demo = gr.Interface(fn=predict,
28
+ inputs=gr.Image(label="Input image", type="pil"),
29
+ outputs="text",
30
+ title="Map identification",
31
+ description="The model predicts whether an image is a map or not. It takes about 30 seconds since it runs on a CPU (it is much faster on a GPU).",
32
+ article="<a href='https://huggingface.co/datasets/sraimund/MapPool'>MapPool - Bubbling up an extremely large corpus of maps for AI</a>")
33
  demo.launch()