Upload app.py

app.py

@@ -41,6 +41,7 @@ import random
 
 def greet(X, ny):
     global eng_dict
+    ny = int(ny)
     if ny == 0:
         rand_no = random.random()
         tok_map = {2: 0.4363429005892416,
@@ -173,27 +174,7 @@ def greet(X, ny):
 #         else:
 #             astr+=mock.capitalize()     
     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:
-            break
-        elif pll >= tot_pll:
-            continue
-        else:
-            add_out = er
-            tot_pll = pll
-    print_str= print_str+fin_out+add_out
-    return print_str
-            
+    return er
 title = "Rename a variable in a Java class"
 description = """This model is a fine-tuned GraphCodeBERT model fine-tuned to output higher-quality variable names for Java classes. Long classes are handled by the 
 model. Replace any variable name with a "[MASK]" to get an identifier renaming.
@@ -219,7 +200,7 @@ ex.printStackTrace();
             }
         }
     }
-}"""], ["""import java.net.*;
+}""", "0"], ["""import java.net.*;
 import java.io.*;
 
 public class s {
@@ -237,7 +218,7 @@ public class s {
             if ([MASK] != null) [MASK].close();
         }
     }
-}"""], ["""import java.io.*;
+}""", "2"], ["""import java.io.*;
 import java.util.*;
 
 public class y {
@@ -254,13 +235,13 @@ public class y {
             e.printStackTrace();
         }
     }
-}"""]]
+}""", "2"]]
 # 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",lines=1)
 
-gr.Interface(title = title, description = description, examples = ex, fn=meet, inputs=[
-        textbox
+gr.Interface(title = title, description = description, examples = ex, fn=greet, inputs=[
+        textbox,textbox1
     ], outputs="text").launch()