Update app.py

app.py

@@ -7,7 +7,7 @@ def cap(match):
     return(match.group().capitalize())
     
 
-def predict(input_text, brakes):
+def predict(brakes, input_text):
     
     model = PunctuationModel()
     output_text = model.restore_punctuation(input_text)
@@ -40,8 +40,7 @@ def predict(input_text, brakes):
         # assemble the array back to a string
         pcnt_file_cr=' '.join(pcnt_file_array_hash).replace('#','\n')
 
-    
-    if 'sentences' in brakes:
+    elif 'sentences' in brakes:
         pass
     
     regex1 = r"\bi\b"
@@ -61,7 +60,7 @@ Model restores punctuation and case i.e. of the following punctuations -- [! ? .
     #examples = [['sentences', "my name is clara i live in berkeley california"]]
 
     interface = gr.Interface(fn = predict,
-                         inputs = [gr.CheckboxGroup(["sentences", "timelines"], label="brakes"),
+                         inputs = [gr.Radio(["sentences", "timelines"], label="brakes"),
                                    "text"],
                          outputs = ["text"],
                          title = title,