Spaces:
Running
Running
sashavor
commited on
Commit
·
87d659a
1
Parent(s):
9e84fe7
cleaning
Browse files
app.py
CHANGED
|
@@ -1,12 +1,9 @@
|
|
| 1 |
import pickle
|
| 2 |
-
|
| 3 |
import gradio as gr
|
| 4 |
from datasets import load_dataset
|
| 5 |
from transformers import AutoModel, AutoFeatureExtractor
|
| 6 |
|
| 7 |
|
| 8 |
-
seed = 42
|
| 9 |
-
|
| 10 |
# Only runs once when the script is first run.
|
| 11 |
with open("index_768_cosine.pickle", "rb") as handle:
|
| 12 |
index = pickle.load(handle)
|
|
@@ -35,9 +32,6 @@ def query(image, top_k=4):
|
|
| 35 |
return images_with_captions, labels_with_probs
|
| 36 |
|
| 37 |
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
with gr.Blocks() as demo:
|
| 42 |
gr.Markdown("# Find my Butterfly 🦋")
|
| 43 |
gr.Markdown("## Use this Space to find your butterfly, based on the [iNaturalist butterfly dataset](https://huggingface.co/datasets/huggan/inat_butterflies_top10k)!")
|
|
|
|
| 1 |
import pickle
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
from datasets import load_dataset
|
| 4 |
from transformers import AutoModel, AutoFeatureExtractor
|
| 5 |
|
| 6 |
|
|
|
|
|
|
|
| 7 |
# Only runs once when the script is first run.
|
| 8 |
with open("index_768_cosine.pickle", "rb") as handle:
|
| 9 |
index = pickle.load(handle)
|
|
|
|
| 32 |
return images_with_captions, labels_with_probs
|
| 33 |
|
| 34 |
|
|
|
|
|
|
|
|
|
|
| 35 |
with gr.Blocks() as demo:
|
| 36 |
gr.Markdown("# Find my Butterfly 🦋")
|
| 37 |
gr.Markdown("## Use this Space to find your butterfly, based on the [iNaturalist butterfly dataset](https://huggingface.co/datasets/huggan/inat_butterflies_top10k)!")
|