Update app.py

app.py

@@ -57,7 +57,7 @@ examples = [
     "Śledził mnie niebieski potwór, ale się nie bałem. Byłem spokojny i zrelaksowany.",
 ]
 
-#############################
+############ Language Test ##############
 
 interface_words = gr.Interface(
         fn=check_lang,
@@ -69,7 +69,31 @@ interface_words = gr.Interface(
         cache_examples=True,
 )
 
-#############################
+############ Anomimise ##############
+
+def anon(text):
+    anon_dreams_list = dreamy.anonimise(
+        [text], 
+        return_original=False, 
+        batch_size=1
+    )
+    
+    return anon_dreams_list[0]
+
+interface_anon = gr.Interface(
+        fn=anon,
+        inputs="text", 
+        outputs="text",
+        title="Anonymise",
+        description=description_main,
+        examples=[
+            "I was an exchange student in Slovakia, visiting someone who had been an exchange student here. The start of the dream was when I had just arrived at their house. Her family was very happy to see me, as well as excited to hear about my life in America. They asked me if I could make them a traditional American dessert to have with dinner, and I agreed, but I needed to go to the store to get some strawberries. The former exchange student, Zelka, said she would drive me to the store, so she took me out to her car which was a jeep like the ones in old WWII movies. The ride was very bumpy and when we got to town, I was very surprised to see how many American stores, etc., were there. It was terrible how much American culture had taken over their country. Then we were taken into this shop where a Chinese man was giving acupuncture lessons.", 
+            "I was with my friends Mike and Ray. Lego blocks broke apart. Were perhaps in a junkyard.", 
+        ],
+        cache_examples=True,
+)
+
+############ SA classification #################
 
 pipe_L = pipeline(
     "text-classification",
@@ -92,7 +116,7 @@ interface_model_L =  gr.Interface(
         cache_examples=True,
 )
 
-#############################
+############ SA classification small ###############
 
 pipe_S = pipeline(
     "text-classification",
@@ -115,15 +139,7 @@ interface_model_S =  gr.Interface(
         cache_examples=True,
 )
 
-#############################
-
-# interface_model_G = gr.Interface.load(
-#             "models/DReAMy-lib/t5-base-DreamBank-Generation-Emot-Char",
-#             examples=examples_g,
-#             title="SA Generation",
-# )
-
-#############################
+########## RE ##############
 
 interface_model_RE = gr.Interface(
         text_to_graph,
@@ -139,7 +155,7 @@ interface_model_RE = gr.Interface(
         cache_examples=True,
 )
 
-#############################
+############ NER  #################
 
 pipe_N = pipeline(
     "text2text-generation",
@@ -151,6 +167,7 @@ def predictN(text):
     t = pipe_N(text)
     t = t[0]["generated_text"]
     return t
+
 interface_model_N =  gr.Interface(
         fn=predictN,
         inputs='text',
@@ -161,9 +178,9 @@ interface_model_N =  gr.Interface(
         cache_examples=True,
 )
 
-#############################
+########### MIx all  ################
 
 gr.TabbedInterface(
-    [interface_words, interface_model_N, interface_model_L, interface_model_S, interface_model_RE], 
-    ["Main", "NER", "SA Multilingual", "SA English", "RE"],
+    [interface_words, interface_anon, interface_model_N, interface_model_L, interface_model_S, interface_model_RE], 
+    ["Main", "Anonymise", "NER", "SA Multilingual", "SA English", "RE"],
 ).launch()