Update app.py

app.py

@@ -30,9 +30,11 @@ model_box=[
 ]
 current_model=model_box[0]
 pythonFlag = "false"
+javaFlag = "false"
 
 def the_process(input_text, model_choice):
     global pythonFlag
+    global javaFlag
     print("Inside the_process for python 0", pythonFlag)
     global output
     print("Inside the_process for python 1", model_choice)
@@ -46,6 +48,16 @@ def the_process(input_text, model_choice):
         pythonFlag = "true"
       elif(pythonFlag == "true"):
         print("pythonFlag", pythonFlag)
+     if(model_choice==0):
+       if(javaFlag == "false"):
+            print("Inside llama for python")
+            tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
+            model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
+            output = run_predict(input_text, model, tokenizer)
+            print("output" , output)
+            javaFlag = "true"
+          elif(javaFlag == "true"):
+            print("javaFlag", javaFlag)
     else:
       a_variable = model_box[model_choice]
       output = a_variable(input_text)