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()
|