Spaces:
Runtime error
Runtime error
import gradio as gr | |
import tensorflow as tf | |
import pandas as pd | |
import numpy as np | |
import json | |
import keras | |
from keras.models import load_model | |
import tensorflow_hub as hub | |
import tensorflow_text as text | |
def make_prediction(tweet): | |
ser = pd.Series([tweet]) | |
x_val = ser.to_frame(name="tweet") | |
x_val = x_val['tweet'] | |
model = load_model("bert_small.h5" ,custom_objects={'KerasLayer': hub.KerasLayer} ) | |
y = model.predict(x_val) | |
return y[0][0] | |
tweet_input = gr.Textbox(label = "Enter the tweet") | |
output = gr.Number(label = "Probability of a True claim") | |
app = gr.Interface(fn = make_prediction, inputs=tweet_input , outputs=output, title="COVID News Ground Reality Predictor") | |
app.launch() | |