from description_validator import Description_Validator | |
import gradio as gr | |
description_validator = Description_Validator(model_name="DistilRoBERTa-v1") | |
iface = gr.Interface(description_validator.similarity_score, | |
inputs=["text", "text"], | |
outputs="number", | |
title="Product Description Similarity Calculator", | |
description="Enter two product descriptions to calculate their similarity." | |
) | |
iface.launch() |