import gradio as gr Model = gr.Interface.load("models/sasi2400/AnswerBERT_model") def prediction(Q,A): input = str(Q).lower() + " " +str(A).lower() model = Model output = model(input) return output gr.Interface(fn=prediction, inputs=["text","text"], outputs=["label"]).launch()#(share=True,debug=True )