Spaces:
Runtime error
Runtime error
Commit
·
ccdb9f2
1
Parent(s):
bfbd3f8
full test2
Browse files
app.py
CHANGED
@@ -1,26 +1,26 @@
|
|
1 |
import gradio
|
2 |
-
import pandas as pd
|
3 |
-
from annoy import AnnoyIndex
|
4 |
-
from deepface import DeepFace
|
5 |
-
import gdown
|
6 |
-
from similarity_finder.get_similar_images import get_similar_images
|
7 |
|
8 |
|
9 |
-
def my_inference_function(image_path):
|
10 |
-
global df
|
11 |
-
r = get_similar_images(df, image_path, age, gender, race)
|
12 |
-
return "Hello " + image_path + "!"
|
13 |
|
14 |
gradio_interface = gradio.Interface(
|
15 |
fn = my_inference_function,
|
16 |
-
inputs = gradio.Image(type="filepath"),
|
17 |
outputs = "text"
|
18 |
)
|
19 |
|
20 |
print("downloading dataset")
|
21 |
-
id = "1z6amrk21M-4N39OZLOsi4WWjAdBVSjh7"
|
22 |
-
gdown.download(id=id, output="final_dataframe.parquet.gzip", quiet=False)
|
23 |
|
24 |
print("loading dataframe")
|
25 |
-
df = pd.read_parquet("final_dataframe.parquet.gzip")
|
26 |
gradio_interface.launch(share=True)
|
|
|
1 |
import gradio
|
2 |
+
#import pandas as pd
|
3 |
+
#from annoy import AnnoyIndex
|
4 |
+
#from deepface import DeepFace
|
5 |
+
#import gdown
|
6 |
+
#from similarity_finder.get_similar_images import get_similar_images
|
7 |
|
8 |
|
9 |
+
def my_inference_function(image_path, age, gender, race):
|
10 |
+
#global df
|
11 |
+
#r = get_similar_images(df, image_path, age, gender, race)
|
12 |
+
return "Hello " + image_path + age + gender + race "!"
|
13 |
|
14 |
gradio_interface = gradio.Interface(
|
15 |
fn = my_inference_function,
|
16 |
+
inputs = [gradio.Image(type="filepath"), gr.Textbox(label="Age"), gr.Textbox(label="Gender"), gr.Textbox(label="Race")],
|
17 |
outputs = "text"
|
18 |
)
|
19 |
|
20 |
print("downloading dataset")
|
21 |
+
#id = "1z6amrk21M-4N39OZLOsi4WWjAdBVSjh7"
|
22 |
+
#gdown.download(id=id, output="final_dataframe.parquet.gzip", quiet=False)
|
23 |
|
24 |
print("loading dataframe")
|
25 |
+
#df = pd.read_parquet("final_dataframe.parquet.gzip")
|
26 |
gradio_interface.launch(share=True)
|