Added classification tasks

app.py

@@ -6,6 +6,8 @@ import apps.summarization
 import apps.home
 import apps.paraphrasing
 import apps.title_generation
+import apps.sentiment
+import apps.categorization
 
 st.set_page_config(
         page_title="Turna",
@@ -18,6 +20,8 @@ PAGES = {
     "Text Summarization": apps.summarization,
     "Text Paraphrasing": apps.paraphrasing,
     "News Title Generation": apps.title_generation,
+    "Sentiment Classification": apps.sentiment,
+    "Text Categorization": apps.categorization,
 }
 
 

apps/categorization.py

@@ -0,0 +1,49 @@
+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("# Text Categorization")
+	st.sidebar.header("Text Categorization")
+	st.write(
+		"""Here, you can perform text categorization using the fine-tuned TURNA classification models. """
+	)
+
+	# Sidebar
+
+	# Taken from https://huggingface.co/spaces/flax-community/spanish-gpt2/blob/main/app.py
+	st.sidebar.subheader("Configurable parameters")
+
+	model_name = st.sidebar.selectbox(
+		"Model Selector",
+		options=[
+			"turna_classification_ttc4900",
+			"turna_classification_ttc4900_NLU",
+			"turna_classification_ttc4900_NLG",
+			"turna_classification_ttc4900_S2S",
+		],
+		index=0,
+	)
+	"""max_new_tokens = st.sidebar.number_input(
+		"Maximum length",
+		min_value=0,
+		max_value=20,
+		value=20,
+		help="The maximum length of the sequence to be generated.",
+	)"""
+	
+	input_text = st.text_area(label='Enter a text: ', height=200, 
+			value=" anadolu_efes e 18 lik star ! beko_basketbol_ligi nde iddialı bir kadroyla sezona giren anadolu_efes transfer harekatına devam ediyor avrupa basınında yer alan iddialara göre lacivert beyazlıların son hedefi kk zagreb de forma giyen 1994 doğumlu dario saric hırvat oyuncunun anadolu_efes ile kesin anlaşmaya vardığı iddia edilirken efes in bu transfer için kk zagreb e 550 bin euro ödeyeceği ifade edildi saric in sezon sonuna kadar şu anda kiralık olarak formasını giydiği kk split te kalacağı ve sezon sonunda anadolu_efes e katılacağı belirtildi hırvat basketbolunun gelecek vaadeden isimlerinden biri olarak gösterilen saric 2 05 boyunda ve kısa forvet pozisyonunda görev yapıyor yıldız basketbolcu 2012 18 yaş altı avrupa_basketbol_şampiyonasında hırvatistan ı şampiyonluğa taşımıştı final karşılaşmasında litvanya potasına 39 sayı bırakan saric turnuvayı 25 6 sayı 10 1 ribaund ve 3 3 asist ortalamasıyla tamamlamıştı")
+	url = ("https://api-inference.huggingface.co/models/boun-tabi-LMG/" + model_name.lower())
+	params = {"max_new_tokens": 8 }
+	if st.button("Generate"):
+		with st.spinner('Generating...'):
+			output = query(input_text, url, params)
+			st.success(output)

apps/sentiment.py

@@ -0,0 +1,53 @@
+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("# Sentiment Classification")
+	st.sidebar.header("Sentiment Classification")
+	st.write(
+		"""Here, you can perform sentiment classification using the fine-tuned TURNA classification models. """
+	)
+
+	# Sidebar
+
+	# Taken from https://huggingface.co/spaces/flax-community/spanish-gpt2/blob/main/app.py
+	st.sidebar.subheader("Configurable parameters")
+
+	model_name = st.sidebar.selectbox(
+		"Model Selector",
+		options=[
+			"turna_classification_17bintweet_sentiment",
+			"turna_classification_17bintweet_sentiment_NLU",
+			"turna_classification_17bintweet_sentiment_NLG",
+			"turna_classification_17bintweet_sentiment_S2S",
+			"turna_classification_tr_product_reviews",
+			"turna_classification_tr_product_reviews_NLU",
+			"turna_classification_tr_product_reviews_NLG",
+			"turna_classification_tr_product_reviews_S2S",
+		],
+		index=0,
+	)
+	"""max_new_tokens = st.sidebar.number_input(
+		"Maximum length",
+		min_value=0,
+		max_value=20,
+		value=20,
+		help="The maximum length of the sequence to be generated.",
+	)"""
+	
+	input_text = st.text_area(label='Enter a text: ', height=100, 
+			value="sonunda bugün kurtuldum senden")
+	url = ("https://api-inference.huggingface.co/models/boun-tabi-LMG/" + model_name.lower())
+	params = {"max_new_tokens": 4 }
+	if st.button("Generate"):
+		with st.spinner('Generating...'):
+			output = query(input_text, url, params)
+			st.success(output)