Spaces:
Runtime error
Runtime error
File size: 5,456 Bytes
23d6e6c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
import streamlit as st
import numpy as np
from keras.utils import pad_sequences
from keras.models import load_model
import pickle
from PIL import Image
image = Image.open('malaria-header.png')
st.image(image)
def predict_labels(input_text):
MAX_NUM_WORDS = 10000
MAX_SEQUENCE_LENGTH = 20
with open(r'tokenizer.pkl', 'rb') as f:
tokenizer = pickle.load(f)
X = tokenizer.texts_to_sequences(input_text.split())
X = pad_sequences(X, maxlen=MAX_SEQUENCE_LENGTH)
label2idx = {'O': 0, 'S-Disease': 1, 'B-Disease': 2, 'B-Organism': 3, 'I-Organism': 4,
'S-Chemical_Substance': 5, 'S-Protien': 6, 'S-Organism': 7, 'B-Chemical_Substance': 8,
'B-Protien': 9, 'I-Disease': 10, 'B-Medication': 11, 'S-Medication': 12,
'S-Anatomical_Substances': 13, 'I-Protien': 14, 'B-Anatomical_Substances': 15,
'I-Chemical_Substance': 16, 'I-Anatomical_Substances': 17, 'I-Medication': 18,
'S-Gene': 19, 'E-Disease': 20, 'E-Protien': 21, 'E-Organism': 22, 'S_Anatomical_Substances': 23,
'B-Gene': 24, 'E-Medication': 25, 'E-Anatomical_Substances': 26, 'E-Chemical_Substance': 27,
'I-Gene': 28, 'E-Gene': 29}
# Load the trained model
model = load_model(r'model.h5')
# Predict on the new data
predictions = model.predict(X)
# Convert predictions to labels
predicted_labels = []
for pred in predictions:
#print(pred)
#predicted_labels.append(list(label2idx.keys())[list(label2idx.values()).index(np.argmax(pred))])
predicted_label_idx = np.argmax(pred)
#print(predicted_label_idx)
predicted_label = list(label2idx.keys())[predicted_label_idx]
predicted_labels.append(predicted_label)
return predicted_labels
#Create the Streamlit app
input_text = st.text_area('Enter some text on Malaria:')
st.write("OR")
option = st.selectbox(
'Choose a text from dropdown: ',
('Malaria remains an important public health problem despite efforts to control it. Besides active transmission, relapsing malaria caused by dormant liver stages of Plasmodium vivax and Plasmodium ovale hypnozoites is a major hurdle in malaria control and elimination programs. Primaquine is the most widely used drug for radical cure of malaria.',
'RTS,S/AS01 (RTS,S) is the first and, to date, only vaccine that has demonstrated it can significantly reduce malaria in young children living in moderate-to-high malaria transmission areas. It acts against the Plasmodium falciparum parasite, the deadliest malaria parasite globally.',
'Malaria is an acute febrile illness caused by Plasmodium parasites, which are spread to people through the bites of infected female Anopheles mosquitoes. It is preventable and curable.'))
# Define the colors for each label
entity_colors = {"O": "white",
"Disease": "#ffa07a",
"Organism": "#98fb98",
"Chemical_Substance": "#1e90ff",
"Protien": "#9370db",
"Medication": "#ffff00",
"Anatomical_Substances": "#ffb6c1",
"Gene": "#87cefa"}
if st.button('Analyze'):
if input_text:
predicted_labels = predict_labels(input_text)
# Split input text into words
words = input_text.split()
# Use a for loop to replace each word with highlighted text using the corresponding predicted label
highlighted_text = ""
for i, word in enumerate(words):
label = predicted_labels[i]
if label != 'O':
label=label.split("-")[1]
word_color = entity_colors.get(label, "yellow")
label_color = entity_colors.get(label + '-label', "<b>black</b>")
highlighted_text += f'<mark style="background-color: {word_color}; color: {label_color}; padding: 0 0.25rem; border-radius: 0.25rem; border: 2px solid {word_color}; border-bottom-width: 1px">{word}<sup style="background-color: white; color: black; border: 1px solid black; border-radius: 2px; padding: 0 0.15rem; font-size: 70%; margin-left: 0.15rem; font-weight: bold;">{label}</sup></mark> '
else:
highlighted_text += f'{word} '
# Display the highlighted text
st.markdown(highlighted_text, unsafe_allow_html=True)
elif option:
predicted_labels = predict_labels(option)
# Split input text into words
words = option.split()
# Use a for loop to replace each word with highlighted text using the corresponding predicted label
highlighted_text = ""
for i, word in enumerate(words):
label = predicted_labels[i]
if label != 'O':
label=label.split("-")[1]
word_color = entity_colors.get(label, "yellow")
label_color = entity_colors.get(label+ '-label', "black")
highlighted_text += f'<mark style="background-color: {word_color}; color: {label_color}; padding: 0 0.25rem; border-radius: 0.25rem; border: 2px solid {word_color}; border-bottom-width: 1px">{word}<sup style="background-color: white; color: black; border: 1px solid black; border-radius: 2px; padding: 0 0.15rem; font-size: 70%; margin-left: 0.15rem; font-weight: bold;">{label}</sup></mark> '
else:
highlighted_text += f'{word} '
# Display the highlighted text
st.markdown(highlighted_text, unsafe_allow_html=True)
|