Added the NLI task

app.py

@@ -11,6 +11,7 @@ import apps.categorization
 import apps.ner
 import apps.pos_tagging
 import apps.sts
+import apps.nli
 import apps.turna
 
 st.set_page_config(
@@ -29,6 +30,7 @@ PAGES = {
     "Named Entity Recognition": apps.ner,
     "Part-of-Speech Tagging": apps.pos_tagging,
     "Semantic Textual Similarity": apps.sts,
+    "Natural Language Inference": apps.nli,
     "Text Generation": apps.turna,
 }
 

apps/nli.py

@@ -0,0 +1,28 @@
+import requests
+import streamlit as st
+import time 
+from transformers import pipeline
+import os
+from .utils import query
+
+HF_AUTH_TOKEN = os.getenv('HF_AUTH_TOKEN')
+headers = {"Authorization": f"Bearer {HF_AUTH_TOKEN}"}
+
+def write():
+
+	st.markdown("# Natural Language Inference")
+	st.sidebar.header("Natural Language Inference")
+	st.write(
+		"""Here, you can perform Natural Language Inference using the fine-tuned TURNA NLI model. """
+	)
+	
+	hypothesis = st.text_area(label='Hypothesis: ', height=50, 
+			value="Bunu nereden biliyorsun? Bütün bunlar yine onların bilgileri.")
+	premise = st.text_area(label='Premise: ', height=50, 
+			value="Bu bilgi onlara ait.")
+	url = ("https://api-inference.huggingface.co/models/boun-tabi-LMG/turna_nli_nli_tr")
+	params = {"max_new_tokens": 8 }
+	if st.button("Generate"):
+		with st.spinner('Generating...'):
+			output = query(f"hipotez: {hypothesis} önerme: {premise}", url, params)
+			st.success(output)