Update models/hsd_tr.py

models/hsd_tr.py

@@ -10,15 +10,17 @@ label_dict = {'LABEL_0': 'Hate', 'LABEL_1': 'Non-hate'}
 
 
 #st.title("Hate Speech Detection in Turkish with HuggingFace Spaces")
-#st.write("Enter a Turkish sentence to detect its hateful:")
-
+#st.write("Enter a Turkish sentence here:")
 #user_input = st.text_input(" ")
-#user_input = st.text_input("Enter your text here:",)
-if st.button("Predict"):
+
+user_input = st.text_input("Enter your text here:",)
+
 #if user_input:
-    result = sentiment_pipeline(user_input)
-    sentiment = result[0]["label"]
-    label_dict = {'LABEL_0': 'Hate', 'LABEL_1': 'Non-hate'}
-    sentiment = label_dict[sentiment]
-    #st.write(f"Detection: {sentiment}")
-    #st.success(sentiment)
+if st.button("Click for predictions!"):
+    with st.spinner('Generating predictions...'):
+        result = sentiment_pipeline(user_input)
+        sentiment = result[0]["label"]
+        label_dict = {'LABEL_0': 'Hate', 'LABEL_1': 'Non-hate'}
+        sentiment = label_dict[sentiment]
+        #st.write(f"Detection: {sentiment}")
+        #st.success(sentiment)