Update app.py
Browse files
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,
|
|
|
|
|
|
|
|
|
|
|
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()
|