import ktrain import gradio as gr predictor = ktrain.load_predictor('bert') def func(text): results = predictor.predict(text) results = [int(element) for element in results] return results gr.Interface(fn=func, inputs="text", outputs="text").launch()