Update app.py

app.py

@@ -1,5 +1,16 @@
-
 import streamlit as st
+from langchain.embeddings import OpenAIEmbeddings
+
+# Load model from HF Spaces
+model = OpenAIEmbeddings(model_name="PyaeSoneK/legalQAcustom") 
+
+# Streamlit UI
+st.header("Seon's Legal QA for Dummies")
+input_text = st.text_input("Give me your wildest legal thoughts:")
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+if st.button("Generate"):
+  # Get embeddings
+  embeddings = model(input_text)
+  
+  # Display results
+  st.write(embeddings)