Update app.py

app.py

@@ -52,6 +52,20 @@ def the_process(input_text, model_choice):
         model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
         output = run_predict(input_text, model, tokenizer)
         print("output starcoder java" , output)
+    elif(model_choice==2):
+      if(javaFlag == "false"):
+        print("Inside starcoder for java")
+        tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/codet5-large-hf-python")
+        model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/codet5-large-hf-python")
+        output = run_predict(input_text, model, tokenizer)
+        print("output starcoder java" , output)
+    elif(model_choice==3):
+      if(javaFlag == "false"):
+        print("Inside starcoder for java")
+        tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/codet5-large-hf")
+        model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/codet5-large-hf")
+        output = run_predict(input_text, model, tokenizer)
+        print("output starcoder java" , output)
     else:
       a_variable = model_box[model_choice]
       output = a_variable(input_text)