mattikris commited on
Commit
24aafee
1 Parent(s): b7cf781

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +42 -0
app.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ st.markdown(""" This is a Streamlit App """)
3
+
4
+ import streamlit as st
5
+ import pandas as pd
6
+ import numpy as np
7
+ import pickle
8
+ import chardet
9
+ from pathlib import Path
10
+ from detect_delimiter import detect
11
+
12
+ label_dict = {
13
+ 0: "Brandsøgning",
14
+ 1: "Informational",
15
+ 2: "Inspiration",
16
+ 3: "Navigational",
17
+ 4: "Transactional"
18
+ }
19
+ upload_file = st.file_uploader("Choose a file",type="csv" )
20
+ model = pickle.load(open("finalized_model.sav","rb"))
21
+
22
+ if upload_file is not None:
23
+ result = chardet.detect(upload_file.getvalue())
24
+ encoding_value = result["encoding"]
25
+ if encoding_value == "UTF-16":
26
+ white_space = True
27
+ else:
28
+ white_space = False
29
+ df = pd.read_csv((upload_file), on_bad_lines='skip', encoding=encoding_value, delim_whitespace=white_space)
30
+ print(df)
31
+ result = {}
32
+ result['Keyword'] = df['Keyword'][:5000]
33
+ result['volume'] =df['Volume'][:5000]
34
+ classes = [label_dict[model.predict(item)[0][0]] for item in df['Keyword'].values[:5000]]
35
+ result['Classes'] = classes
36
+ df = pd.DataFrame(result)
37
+ st.download_button(
38
+ label="Download CSV file",
39
+ data=df.to_csv().encode('utf-8'),
40
+ file_name='labbeled_data.csv',
41
+ mime='text/csv'
42
+ )