change charts to plotly

app.py

@@ -482,6 +482,7 @@ from transformers import AutoConfig
 from transformers import AutoTokenizer
 import pandas as pd
 import openai
+import plotly.express as px
 
 model_ckpt = 'bert-base-uncased'
 tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
@@ -503,14 +504,18 @@ EMBEDDING_ENCODING = 'cl100k_base'
 king = openai.Embedding.create(input=token_king, model=EMBEDDING_MODEL)["data"][0]["embedding"]
 queen = openai.Embedding.create(input=token_queen, model=EMBEDDING_MODEL)["data"][0]["embedding"]
 
-st.write("Google's 'bert-base-uncased' model embeddings:")
 df = pd.DataFrame({f'"{token_king}" embeddings': king_emb_np[:50], f'"{token_queen}" embeddings': queen_emb_np[:50]})
-st.line_chart(df)
+fig = px.line(df, title="Google's 'bert-base-uncased' model embeddings")
+fig.update_layout(legend=dict(orientation="h"))
+st.plotly_chart(fig, use_container_width=True)
 
 
-st.write("OpenAI's 'text-embedding-ada-002' model embeddings:")
 df = pd.DataFrame({f'"{token_king}" embeddings': king[:50], f'"{token_queen}" embeddings': queen[:50]})
-st.line_chart(df)
+fig = px.line(df, title="OpenAI's 'text-embedding-ada-002' model embeddings")
+fig.update_layout(legend=dict(orientation="h"))
+st.plotly_chart(fig, use_container_width=True)
+
+
 
 
 

requirements.txt

@@ -2,4 +2,5 @@ streamlit~=1.21.0
 tokenizers~=0.13.3
 transformers~=4.31.0
 torch~=2.0.1
-openai~=0.27.8
+openai~=0.27.8
+plotly~=5.15.0