99ashutosh commited on
Commit
ccdb9f2
·
1 Parent(s): bfbd3f8

full test2

Browse files
Files changed (1) hide show
  1. app.py +13 -13
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)