app.py

import gradio
import pandas as pd
from annoy import AnnoyIndex
from deepface import DeepFace
import gdown
from similarity_finder.get_similar_images import get_similar_images


def my_inference_function(image_path, age, gender, race):
  global df
  r = get_similar_images(df, image_path, int(age), gender, race)
  return r

gradio_interface = gradio.Interface(  
  fn = my_inference_function,
  inputs = [gradio.Image(type="filepath"), gradio.Textbox(label="Age"), gradio.Textbox(label="Gender"), gradio.Textbox(label="Race")],
  outputs = "text"
)

print("downloading dataset")
id = "1z6amrk21M-4N39OZLOsi4WWjAdBVSjh7"
gdown.download(id=id, output="final_dataframe.parquet.gzip", quiet=False)

print("loading dataframe")
df = pd.read_parquet("final_dataframe.parquet.gzip")
gradio_interface.launch(share=True)