Update app.py

app.py

@@ -147,7 +147,7 @@ database_df = test_dataset.load_database()
 database_embeddings = encode_database(model, database_df)
 
 
-def interface_fn(selected_image, query_text):
+def interface_fn(selected_image: str, query_text: str) -> Image.Image:
     if selected_image is None:
         return None
         
@@ -156,7 +156,7 @@ def interface_fn(selected_image, query_text):
         selected_image, 
         query_text, 
         database_embeddings, 
-        database_df  # Pass the pre-loaded DataFrame
+        database_df
     )
     return Image.open(result_image_path)