Update app.py

app.py

@@ -125,11 +125,6 @@ def main(args):
 def make_frontend(fn: Callable[[str], List[str]], app_name: str = "fashion-aggregator"):
     """Creates a gradio.Interface frontend for text to image search function."""
 
-    def preprocess_input(input_str):
-        paths_and_scores = fn(input_str)
-        images = [gr.utils.image.load_image(path) for path in paths_and_scores]
-        return images
-
     input_section = gr.inputs.Textbox(lines=2, placeholder="Enter item description here")
 
     output_section = gr.Gallery(label="Relevant Items", show_labels=True, label_position="below")
@@ -158,7 +153,7 @@ def make_frontend(fn: Callable[[str], List[str]], app_name: str = "fashion-aggre
 
     # Build a customized browser interface to a Python function
     frontend = gr.Interface(
-        fn=preprocess_input,
+        fn=fn,
         inputs=input_section,
         outputs=output_section,
         layout=interface_layout,