yugamj's picture
removed pip install commands
745fb2d
raw
history blame
591 Bytes
#Import and install libraries
import numpy as np
import tensorflow as tf
from tensorflow.keras.models import load_model
import tensorflow_text as tf_text
import tensorflow_hub as hub
import gradio as gr
#Import model
model = tf.keras.models.load_model('detect_LLM_colab_method2')
#Function that predicts for a given text
def LLM_text(Text):
prediction = model.predict([Text])
output_text = 'Your text is {}% likely to be a LLM generated.'.format(round(prediction[0][0]*100, 2))
return output_text
iface = gr.Interface(fn=LLM_text, inputs="text", outputs="text")
iface.launch()