Update app.py

app.py

@@ -175,23 +175,29 @@ def greet(X, ny):
     er = er+" (with PLL value of: "+str(vali)+")"
     return er, vali
 
-def meet(X):
-    tot_pll = 100.00
-    print_str = ""
-    fin_out = "The highest confidence prediction is: "
-    add_out = ""
-    for r in range(6):
-        er, pll = greet(X, 6-r)
-        print_str+= er
-        print_str+='\n'
-        if (pll - tot_pll) > 0.1 and tot_pll < 1:
-            break
-        elif pll >= tot_pll:
-            continue
-        else:
-            add_out = er
-            tot_pll = pll
-    print_str= print_str+fin_out+add_out
+def meet(X, ni):
+    ni = int(ni)
+    if ni == 0:
+        print_str,vali = greet(X,ni)
+    elif ni == -1:
+        tot_pll = 100.00
+        print_str = ""
+        fin_out = "The highest confidence prediction is: "
+        add_out = ""
+        for r in range(6):
+            er, pll = greet(X, 6-r)
+            print_str+= er
+            print_str+='\n'
+            if (pll - tot_pll) > 0.1 and tot_pll < 1:
+                break
+            elif pll >= tot_pll:
+                continue
+            else:
+                add_out = er
+                tot_pll = pll
+        print_str= print_str+fin_out+add_out
+    else:
+        print_str,vali = greet(X,ni)
     return print_str
             
 title = "Rename a variable in a Java class"
@@ -219,7 +225,7 @@ ex.printStackTrace();
             }
         }
     }
-}"""], ["""import java.net.*;
+}""", -1], ["""import java.net.*;
 import java.io.*;
 
 public class s {
@@ -237,7 +243,7 @@ public class s {
             if ([MASK] != null) [MASK].close();
         }
     }
-}"""], ["""import java.io.*;
+}""", -1], ["""import java.io.*;
 import java.util.*;
 
 public class y {
@@ -254,13 +260,13 @@ public class y {
             e.printStackTrace();
         }
     }
-}"""]]
+}""", -1]]
 # We instantiate the Textbox class
 textbox = gr.Textbox(label="Type Java code snippet:", placeholder="replace variable with [MASK]", lines=10)
-#textbox1 = gr.Textbox(label="Number of tokens in name:", placeholder="0 for randomly sampled number of tokens",lines=1)
+textbox1 = gr.Textbox(label="Number of tokens in name:", placeholder="0 for randomly sampled number of tokens and -1 for automatic number of token selection",lines=1)
 
 gr.Interface(title = title, description = description, examples = ex, fn=meet, inputs=[
-        textbox
+        textbox, textbox1
     ], outputs="text").launch()