Spaces:
Runtime error
Runtime error
| import os | |
| from laserembeddings import Laser | |
| from scipy import spatial | |
| from transformers import pipeline | |
| import gradio as gr | |
| model = pipeline('question-answering') | |
| os.system("python -m laserembeddings download-models") | |
| laser = Laser() | |
| def question_answer(draft, reference, question): | |
| ref_answer = model({'question': question, 'context': reference})['answer'] | |
| draft_answer = model({'question': question, 'context': draft})['answer'] | |
| embs = laser.embed_sentences([ref_answer, draft_answer], lang='en') | |
| sim = 1 - spatial.distance.cosine(embs[0], embs[1]) | |
| if sim >= 0.8: | |
| out = "🎉 Great draft. No comprehensibility warning." | |
| else: | |
| out = "⚠️ Potential comprehensibility issue!" | |
| return out | |
| iface = gr.Interface(fn=question_answer, inputs=["text", "text", "text"], outputs=["text"]).launch() | |