Update ui_utils.py

ui_utils.py

@@ -72,16 +72,18 @@ def display_query_input(client, selected_class, tokenizer, model):  # Added para
 
 def query_tapas_with_weaviate_data(client, selected_class, question, tokenizer, model):
     # Embed the question
-    question_embedding = tapas_utils.embed_table(pd.DataFrame([question], columns=["Question"]))
+    question_embedding = tapas_utils.embed_question(question)
+    
     # Retrieve the most relevant table
     table = weaviate_utils.retrieve_relevant_table(client, selected_class, question_embedding)
-    # Convert the table to TAPAS format
-    table = weaviate_utils.convert_to_tapas_format(table)
+    
     # Call TAPAS with the table and the question
     answers = tapas_utils.ask_llm_chunk(tokenizer, model, table, [question])
+    
     # Display the answers
     for answer in answers:
         st.write(f"Answer: {answer}")
 
 
 
+