marksverdhei commited on
Commit
74e9a4d
·
1 Parent(s): e8bfa89

Fix kwargs and such

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  import torch
3
  import numpy as np
4
  import views
5
- from resources import load_corrector, load_data, load_model_and_tokenizer
6
 
7
  use_cpu = not torch.cuda.is_available()
8
  device = "cpu" if use_cpu else "cuda"
@@ -19,12 +19,18 @@ def load_embeddings():
19
  return np.load("syac-title-embeddings.npy")
20
 
21
  embeddings = load_embeddings()
22
-
23
 
24
  tab1, tab2 = st.tabs(["plot", "diffs"])
25
 
26
  with tab1:
27
- views.plot()
 
 
 
 
 
 
28
 
29
  with tab2:
30
- views.diffs()
 
2
  import torch
3
  import numpy as np
4
  import views
5
+ from resources import load_corrector, load_data, load_model_and_tokenizer, reduce_embeddings
6
 
7
  use_cpu = not torch.cuda.is_available()
8
  device = "cpu" if use_cpu else "cuda"
 
19
  return np.load("syac-title-embeddings.npy")
20
 
21
  embeddings = load_embeddings()
22
+ vectors_2d, reducer = reduce_embeddings(embeddings)
23
 
24
  tab1, tab2 = st.tabs(["plot", "diffs"])
25
 
26
  with tab1:
27
+ views.plot(
28
+ df=df,
29
+ embeddings=embeddings,
30
+ vectors_2d=vectors_2d,
31
+ reducer=reducer,
32
+ corrector=corrector,
33
+ )
34
 
35
  with tab2:
36
+ views.diffs(embeddings, corrector)