Update README.md

README.md

@@ -5,70 +5,9 @@ datasets:
 language:
 - ka
 ---
-```python
-    import gradio as gr
-    from transformers import pipeline
-    fill_mask = pipeline('fill-mask',
-                     model='davmel/ka_homonym_disambiguation_FM',
-                     tokenizer='davmel/ka_homonym_disambiguation_FM')  # 1788?
+<script
+	type="module"
+	src="https://gradio.s3-us-west-2.amazonaws.com/4.18.0/gradio.js"
+></script>
 
-    masker = {0: 'თო', 1: 'დაბ', 2: 'კაფე'}
-
-    def getMasked(sent):
-        homonym_forms = [
-            'ბარით', 'ბარიანის', 'ბარზე', 'ბარები', 'ბარისთვისაც', 'ბარებია', 'ბარეთსაც', 'ბარაინში', 'ბარეთმაც',
-            'ბარიანი',
-            'ბარიცაა', 'ბარიდანა', 'ბარიც', 'ბარვაზედ', 'ბარელიც', 'ბარავდი', 'ბარისკაცები', 'ბარვაში', 'ბარვას',
-            'ბარეთისა', 'ბარავოოო', 'ბარისებური', 'ბარებისთვისა', 'ბარისათვის', 'ბარზეა', 'ბარელი', 'ბარშივე',
-            'ბარისაა',
-            'ბარისკენ', 'ბარელებით', 'ბარადა', 'ბარვაა', 'ბარს', 'ბარიო', 'ბარავდეს', 'ბარებმა', 'ბარებით', 'ბარების',
-            'ბარისგან', 'ბარშიო', 'ბარებისკენ', 'ბარიდანაა', 'ბარებინა', 'ბარებისა', 'ბარვისგან', 'ბარვისთან',
-            'ბარავდნენ',
-            'ბარადაც', 'ბარელამდე', 'ბარშიაო', 'ბარადაა', 'ბარითა', 'ბარელებიც', 'ბარელის', 'ბარელებს', 'ბარვის',
-            'ბარელებმა', 'ბარელმა', 'ბარადო', 'ბარებისგან', 'ბარვისას', 'ბარებშია', 'ბარავენ', 'ბარვით', 'ბარიანს',
-            'ბარელებში', 'ბარიცო', 'ბარავდა', 'ბარშია', 'ბარვაზე', 'ბარავთ', 'ბარო', 'ბარელისა', 'ბარაინი',
-            'ბარელისათვის',
-            'ბარელია', 'ბარებშიო', 'ბარის', 'ბარისთვისო', 'ბარებად', 'ბარშიგან', 'ბარებსაც', 'ბარებითდა', 'ბარებზე',
-            'ბარია', 'ბარებიდან', 'ბარისად', 'ბარზეც', 'ბარშიც', 'ბარებითაა', 'ბარად', 'ბარიდანაც', 'ბარისმაგვარ',
-            'ბარში',
-            'ბარებისთვისაც', 'ბართანა', 'ბარშიცაა', 'ბარისვე', 'ბარისთვისა', 'ბარმაც', 'ბარელით', 'ბარვა', 'ბარელებთან',
-            'ბარელებზე', 'ბარისაც', 'ბარისებურ', 'ბარელების', 'ბარიდან', 'ბარისი', 'ბარებმაც', 'ბარივით', 'ბარიანმა',
-            'ბარელისთვის', 'ბარებს', 'ბარ', 'ბარებიც', 'ბარებზეც', 'ბარებისთვის', 'ბარელსაც', 'ბარელები', 'ბარისია',
-            'ბარისპირით', 'ბარაამდეე', 'ბარამდეც', 'ბარი', 'ბარისაკენ', 'ბარელად', 'ბარელ', 'ბარამდე', 'ბარიანად',
-            'ბარებისათვის', 'ბარიანია', 'ბარებში', 'ბართან', 'ბარეთთან', 'ბარისაგან', 'ბარებთაგან', 'ბარისპირის',
-            'ბარზედა',
-            'ბარვაო', 'ბარებშიდაა', 'ბარებსადა', 'ბარელამდეც', 'ბარებითა', 'ბარისა', 'ბარებშიც', 'ბარიდანვე', 'ბარისო',
-            'ბარიდანნ', 'ბარელს', 'ბარელთა', 'ბართა', 'ბართანაც', 'ბარავო', 'ბარისთვის', 'ბარელზეა', 'ბარებსა',
-            'ბარისაო',
-            'ბარებიზ���', 'ბარავს', 'ბარისმაგვარი', 'ბარებთან', 'ბარებითო', 'ბარსა', 'ბარული', 'ბარსაც', 'საბარო',
-            'ბარებითო',
-            'ბარშიაც', 'ბარობა', 'ბარისას',
-        ]
-        new_sent = []
-        wasnt = True
-        for word in sent.split(" "):
-            if word in homonym_forms and wasnt:
-                new_sent.append("[MASK]")
-                # wasnt = False
-            else:
-                new_sent.append(word)
-        return " ".join(new_sent)
-
-    def predict(text):
-        # Mask some words in the input text
-        text_with_mask = getMasked(text)
-        # Use the model to predict the masked words
-        predictions = fill_mask(text_with_mask)
-        # Extract the predicted word and score
-        context = predictions[0]["token_str"]
-        if context == masker[0]:
-            return "ომონიმი გამოყენებულია, როგორც იარაღი"
-        elif context == masker[1]:
-            return "ომონიმი გამოყენებულია, როგორც დაბლობი"
-        else:
-            return "ომონიმი გამოყენებულია, როგორც კაფე"
-
-    # Create the Gradio interface
-    iface = gr.Interface(fn=predict, inputs="text", outputs="text", title="Fill-Mask Model",
-                         description="Enter a sentence with a [MASK] token to indicate where the model should fill in the blank. Click 'Submit' to see the model's prediction for the missing word.")
-    iface.launch()
+<gradio-app src="https://davmel-georgian-homonym-disambiguation.hf.space"></gradio-app>