Create app.py

app.py

@@ -0,0 +1,127 @@
+from styleformer import Styleformer
+import torch
+import warnings
+
+warnings.filterwarnings("ignore")
+import gradio as gr
+
+choices = ['Informal to Formal', 'Formal to Informal','Active to passive','Passive to active']
+
+
+def para1(choices, source_sentences):
+    if choices == "Informal to Formal":
+        sf = Styleformer(style=0)
+        sentance1 = list(source_sentences.split("."))
+        output_sentance = []
+        for source_sentence in sentance1:
+            target_sentence = sf.transfer(source_sentence)
+            if target_sentence is not None:
+                output_sentance.append(target_sentence)
+                #print(target_sentence)
+            else:
+                output_sentance.append(target_sentence)
+                #print(target_sentence)
+                output_sentance.append(target_sentence)
+        res = [i for i in output_sentance if i is not None]
+        #print(output_sentance)
+        #print(res)
+        final = ""
+        for value in res:
+            joint_value = "".join(value)
+            if final == "":
+                final += joint_value
+            else:
+                final = f"{final}.{joint_value}"
+        final = final.replace("..", ".")
+        #print(final)
+
+        return final
+
+    if choices == "Formal to Informal":
+        sf = Styleformer(style=1)
+        sentance1 = list(source_sentences.split("."))
+        output_sentance = []
+        for source_sentence in sentance1:
+            target_sentence = sf.transfer(source_sentence)
+            if target_sentence is not None:
+                output_sentance.append(target_sentence)
+                #print(target_sentence)
+            else:
+                output_sentance.append(target_sentence)
+                #print(target_sentence)
+                output_sentance.append(target_sentence)
+        res = [i for i in output_sentance if i is not None]
+        #print(output_sentance)
+        #print(res)
+        final = ""
+        for value in res:
+            joint_value = "".join(value)
+            if final == "":
+                final += joint_value
+            else:
+                final = f"{final}.{joint_value}"
+        final = final.replace("..", ".")
+        #print(final)
+        return final
+    
+    if choices == "Active to passive":
+        sf = Styleformer(style=2)
+        sentance1 = list(source_sentences.split("."))
+        output_sentance = []
+        for source_sentence in sentance1:
+            target_sentence = sf.transfer(source_sentence)
+            if target_sentence is not None:
+                output_sentance.append(target_sentence)
+                #print(target_sentence)
+            else:
+                output_sentance.append(target_sentence)
+                #print(target_sentence)
+                output_sentance.append(target_sentence)
+        res = [i for i in output_sentance if i is not None]
+        #print(output_sentance)
+        #print(res)
+        final = ""
+        for value in res:
+            joint_value = "".join(value)
+            if final == "":
+                final += joint_value
+            else:
+                final = f"{final}.{joint_value}"
+        final = final.replace("..", ".")
+        #print(final)
+        return final
+
+
+    if choices == "Passive to active":
+        sf = Styleformer(style=3)
+        sentance1 = list(source_sentences.split("."))
+        output_sentance = []
+        for source_sentence in sentance1:
+            target_sentence = sf.transfer(source_sentence)
+            if target_sentence is not None:
+                output_sentance.append(target_sentence)
+                #print(target_sentence)
+            else:
+                output_sentance.append(target_sentence)
+                #print(target_sentence)
+                output_sentance.append(target_sentence)
+        res = [i for i in output_sentance if i is not None]
+        #print(output_sentance)
+        #print(res)
+        final = ""
+        for value in res:
+            joint_value = "".join(value)
+            if final == "":
+                final += joint_value
+            else:
+                final = f"{final}.{joint_value}"
+        final = final.replace("..", ".")
+        #print(final)
+        return final
+
+input_1 = gr.inputs.Radio(choices=choices, label='Choose a model.')
+input_2 = gr.inputs.Textbox(placeholder='Enter your text here...', label='Input')
+
+iface = gr.Interface(para1, [input_1, input_2], "text", theme='huggingface')
+if __name__ == "__main__":
+    iface.launch(debug=True)