refac

app.py

@@ -75,19 +75,20 @@ def calculate_embedding(
         embedding = infer(image)
         database.append((img_path, embedding))
     print(f"Embedding Calculation: {time.time() - start}")
-    filepath = "database.npy"
+    filepath = "database.pickle"
     with open(filepath, "wb") as f:
         pickle.dump(database, f)
     return filepath
 
 
 def instance_recognition(
-    embedding_filepath: str,
+    embedding_file,
+    zipfile,
     image_path: str,
 ) -> List[np.ndarray]:
-    with open(embedding_filepath.name, "rb") as f:
+    with open(embedding_file.name, "rb") as f:
         embeddings = pickle.load(f)
-
+    unzip(zipfile.name)
     embedding_vectors = []
     image_paths = []
     for img_path, embedding in embeddings:
@@ -132,6 +133,7 @@ with gr.Blocks() as demo:
             infer_btn = gr.Button(value="Inference")
             input_image = gr.Image(type="filepath", label="Input Image")
             image_embedding_file = gr.File(type="file", label="Image Embeddings")
+            image_zip_file = gr.File(type="file", label="Image Zip File")
 
         with gr.Row():
             output_images = [
@@ -143,7 +145,7 @@ with gr.Blocks() as demo:
 
         infer_btn.click(
             instance_recognition,
-            inputs=[image_embedding_file, input_image],
+            inputs=[image_embedding_file, image_zip_file, input_image],
             outputs=output_images + distances,
         )