Update app.py

app.py

@@ -30,7 +30,7 @@ def reduce_to_3d(embedding):
 
 @spaces.GPU
 def compare_embeddings(*texts):
-    embeddings = [get_embedding(text) for text in texts]
+    embeddings = [get_embedding(text) for text in texts if text.strip()]  # Only process non-empty texts
     embeddings_3d = [reduce_to_3d(emb) for emb in embeddings]
     
     fig = go.Figure()
@@ -51,7 +51,7 @@ def compare_embeddings(*texts):
     
     return fig
 
-def update_interface(num_inputs):
+def create_interface(num_inputs):
     with gr.Blocks() as new_interface:
         text_inputs = [gr.Textbox(label=f"Text {i+1}") for i in range(num_inputs)]
         output = gr.Plot()
@@ -64,8 +64,14 @@ with gr.Blocks() as iface:
     gr.Markdown("Compare the embeddings of multiple strings visualized in 3D space using Mistral 7B.")
     
     num_inputs = gr.Slider(minimum=2, maximum=10, step=1, value=2, label="Number of texts to compare")
-    dynamic_interface = gr.HTML()
+    interface_container = gr.HTML()
     
-    num_inputs.change(fn=update_interface, inputs=[num_inputs], outputs=[dynamic_interface])
+    def update_interface(num):
+        return create_interface(num)
+    
+    num_inputs.change(fn=update_interface, inputs=[num_inputs], outputs=[interface_container])
+    
+    # Initialize the interface with 2 text boxes
+    interface_container.update(create_interface(2))
 
 iface.launch()