File size: 488 Bytes
517a261
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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()