File size: 894 Bytes
3633936
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
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))
    print(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', output='text', title=title, description=description).launch()