Spaces:
Runtime error
Runtime error
File size: 904 Bytes
3633936 b5d8346 3633936 ce9d0ce 3633936 6098534 3633936 d3842c9 |
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 |
from tensorflow.keras.preprocessing.text import Tokenizer
from tensorflow.keras.preprocessing import sequence
from tensorflow.keras.models import load_model
import numpy as np
import gradio as gr
model = load_model('trained.h5')
tokenizer = Tokenizer()
def encoder(text):
text = tokenizer.texts_to_sequences([text])
text = sequence.pad_sequences(text, maxlen=200)
return text
def predict(text):
encoded_text = encoder(text)
#print(encoded_text)
prediction = (model.predict(encoded_text))
return prediction
#prediction = np.round(prediction)
#if prediction==1:
# return "Disaster"
#return "Not a Disaster"
title="Relevance Classifier"
description="<p style='text-align:center'>Classifies input text into Disaster-related or not disaster related."
gr.Interface(fn=predict, inputs='text', outputs='text', title=title, description=description).launch() |