test3 / app.py
flyboytarantino14's picture
Upload 2 files
5795cc4
raw
history blame contribute delete
824 Bytes
#import os
#os.system("python -m pip install --upgrade pip")
#os.system("pip uninstall -y gradio")
#os.system("pip install gradio==3.28.3")
import gradio as gr
from sentence_transformers import SentenceTransformer, util
model = SentenceTransformer('all-MiniLM-L6-v2')
def check_similarity(actual_answer, user_answer):
em_1 = model.encode(actual_answer)
em_2 = model.encode(user_answer)
cosine_scores = util.cos_sim(em_1, em_2)
result = cosine_scores.item()
result = result * 100
percentage = "{:.0f}".format(result)
return percentage
input_actual_answer = gr.Textbox()
input_user_answer = gr.Textbox()
output_percentage = gr.Textbox()
interface = gr.Interface(
fn=check_similarity,
inputs=[input_actual_answer, input_user_answer],
outputs=output_percentage
)
interface.launch()