Spaces:
Sleeping
Sleeping
| #Import all neccessary libraries | |
| import streamlit as st | |
| import re | |
| import wikipediaapi | |
| import malaya | |
| import torch | |
| import tensorflow | |
| import pandas as pd | |
| from sklearn.preprocessing import OneHotEncoder, LabelEncoder | |
| from sklearn.model_selection import train_test_split | |
| from sklearn.neighbors import KNeighborsClassifier | |
| from sklearn.multioutput import MultiOutputClassifier | |
| from sklearn.metrics import classification_report, confusion_matrix, accuracy_score | |
| import numpy as np | |
| import matplotlib.pyplot as plt | |
| import os | |
| import psutil | |
| import pickle | |
| from malaya_file import * | |
| #Page header, title | |
| st.set_page_config(page_title= "Malay Named Entity Recognition (NER) Model", page_icon= ":book:", layout= "wide") | |
| st.title(":book: Pengecaman Entiti Nama Malay (NER) model") | |
| st.markdown("CARA MENGGUNAKAN PROGRAM") | |
| st.markdown("1. Sila taip sebuah ayat atau teks tidak melebihi 500 karakter di ruangan bawah") | |
| st.markdown("2. Pilih model untuk melakukan proses pengecaman entiti nama (NER) berdasarkan teks") | |
| st.markdown("3. Klik butang 'BUAT RAMALAN' bagi memulakan program") | |
| st.markdown("4. Paparan bagi setiap kata serta jenis entiti akan dipaparkan pada bahagian 'HASIL RAMALAN'") | |
| #LOAD MODEL FROM KNN_MALAYA.SAV | |
| st.cache(allow_output_mutation=True) | |
| def knn_malaya(): | |
| load_model = pickle.load(open('knn_malaya.sav', 'rb')) | |
| return load_model | |
| #CHANGE STRING INTO LABELENCODER DATAFRAME | |
| #PREDICT WORD OUTSIDE DATA | |
| st.cache(allow_output_mutation=True) | |
| def tukar_kata(kata): | |
| #global kelas, output | |
| global output | |
| string = re.sub("[=(),:;.]", "", kata) | |
| string1 = string.split(" ") | |
| string2 = pd.DataFrame(string1, columns = ["LKATA"]) | |
| string2['LSEBELUM'] = string2['LKATA'].shift(1) | |
| string2['LSELEPAS'] = string2['LKATA'].shift(-1) | |
| string2.fillna("null", inplace=True) | |
| #string1 | |
| #st.table(string1[:10]) | |
| lbl = LabelEncoder() | |
| lbl_sen = lbl.fit_transform(string2['LKATA']) | |
| lbl_bef = lbl.fit_transform(string2['LSEBELUM']) | |
| lbl_aft = lbl.fit_transform(string2['LSELEPAS']) | |
| string2 = pd.DataFrame({'LKATA':lbl_sen, 'LSEBELUM': lbl_bef, 'LSELEPAS' : lbl_aft}) | |
| #st.dataframe(string2.head()) | |
| #Using classifier.predict() from load_model from knn_malaya.sav | |
| kelas = knn_malaya() | |
| hasil = kelas.predict(string2) | |
| #Convert array into entities | |
| fin = [] | |
| for z in hasil: | |
| if (z == [1, 0, 0]).all(): | |
| fin.append("LOKASI") | |
| elif (z == [0, 1, 0]).all(): | |
| fin.append("MANUSIA") | |
| elif (z == [0, 0, 1]).all(): | |
| fin.append("ORGANISASI") | |
| else: | |
| fin.append("LAIN-LAIN") | |
| #global perkata, output | |
| #perkata = [(key, value) for i, (key, value) in enumerate(zip(string1, fin))] | |
| output = pd.DataFrame({"kata" : string1, "entiti" : fin}) | |
| return output | |
| #CREATE TEXT FORM | |
| with st.form(key= 'my_form'): | |
| global kata, btn_model, df1, df2 | |
| kata = st.text_area(label="Sila taip teks atau ayat:", max_chars= 500) | |
| btn_model = st.radio("Pilih model untuk pengecaman entiti nama", | |
| ("KNN", "BERT", "Tiny-BERT", "ALBERT", "Tiny-ALBERT", "XLNET", "ALXLNET", "FASTFORMER", "Tiny-FASTFORMER")) | |
| submit_button = st.form_submit_button(label= ":arrow_right: Buat Ramalan") | |
| if submit_button: | |
| if re.sub(r'\s+','',kata)=='': | |
| st.error('Ruangan teks tidak boleh kosong.') | |
| elif re.match(r'\A\s*\w+\s*\Z', kata): | |
| st.error("Teks atau ayat mestilah sekurang-kurangnya dua patah perkataan.") | |
| else: | |
| if btn_model == "KNN": | |
| st.write("Anda pilih model : KNN") | |
| result = knn_malaya() | |
| #st.write(load_model) | |
| #df2 = ramal_kata(kata) | |
| else: | |
| st.write("Anda pilih model transformer: ", btn_model) | |
| st.success("Butang hantar berfungsi!") | |
| with st.container(): | |
| st.write("---") | |
| st.header("Hasil Ramalan") | |
| st.subheader("Ayat asal") | |
| st.write("##") | |
| st.write(kata) | |
| patah = str(len(kata.split())) | |
| st.write("Bilangan perkataan : {}".format(patah)) | |
| st.write("##") | |
| if btn_model == 'KNN': | |
| df_test = tukar_kata(kata) | |
| #data = {'Name': ['Tom', 'nick', 'krish', 'jack'], 'entiti': [20, 21, 19, 18]} | |
| #df_test = pd.DataFrame(data) | |
| #st.write("LOAD PICKLE IN PROGRESS") | |
| else: | |
| df_test = malaya_model(btn_model, kata) | |
| entiti = sorted(df_test['entiti'].unique()) | |
| pilih = st.multiselect('Jenis entiti', entiti, entiti) | |
| df_pilihan = df_test [ (df_test['entiti'].isin(pilih)) ] | |
| df_line = df_pilihan.copy() | |
| st.table(df_line.style.set_properties(**{'background-color': 'white', 'color': 'black'})) | |
| #About model | |
| with st.expander("About this app", expanded=True): | |
| st.write( | |
| """ | |
| - **Pengecaman Nama Entiti Malay** adalah sebuah aplikasi pembelajaran mesin yang dibangunkan bagi mengecam entiti pada setiap token menggunakan modul MALAYA (Husein, 2018) | |
| - Projek ini adalah tugasan Final Year Project bagi Ijazah Sarjana Muda di UKM | |
| - Aplikasi ini ingin menentukan model terbaik yang boleh digunakan bagi dokumen teks subjek sejarah Bahasa Melayu | |
| - Model ini mempunyai 3 fitur utama iaitu kata asal, kata sebelum dan kata selepas. Kelas yang disasarkan ialah LOKASI, MANUSIA dan ORGANISASI | |
| - Maklumat lanjut boleh hubungi Muhd Arif Syamil bin Mohd Rahimi melalui e-mel a177313@siswa.ukm.edu.my atau 012-7049021 | |
| """) | |
| process = psutil.Process(os.getpid()) | |
| mem_size = str((process.memory_info().rss)) # in bytes, divide by 1 billion to GB | |
| mem_size_mb = str((process.memory_info().rss) / 1000000) | |
| mem_size_gb = str((process.memory_info().rss) / 1000000000) | |
| st.write("Penggunaan memori: {} bytes or {} MB or {} GB".format(mem_size, mem_size_mb, mem_size_gb)) | |
| # Dokumen Pemasyhuran Kemerdekaan 1957 telah ditulis dalam dua bahasa iaitu bahasa Melayu yang ditulis dalam Jawi dan bahasa Inggeris - No PERSON, LOCATION, ORGANISATION | |
| # Ketika mendarat di Lapangan Terbang Sungai Besi, tetamu kehormat telah disambut oleh Pesuruhjaya Tinggi British di Tanah Melayu, Sir Donald Charles MacGillivray dan Lady MacGillivray, Yang di-Pertuan Agong Tanah Melayu yang pertama, Tuanku Abdul Rahman diiringi Raja Permaisuri Agong dan Perdana Menteri Tanah Melayu yang pertama, Tunku Abdul Rahman. | |
| # Kedudukan sebuah kereta yang terjunam ke dalam Sungai Maaw di Jeti Feri Tanjung Kunyit, Sibu, semalam, sudah dikenal pasti. Jurucakap Pusat Gerakan Operasi (PGO), Jabatan Bomba dan Penyelamat Malaysia (JBPM) Sarawak, berkata kedudukan Toyota Camry di dasar sungai itu dikesan anggota Pasukan Penyelamat Di Air (PPDA) yang melakukan selaman kelima, hari ini, pada jam 3.49 petang. | |