Update app.py

app.py

@@ -44,23 +44,24 @@ def the_process(input_text, model_choice):
         tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf_python")
         model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf_python")
         output = run_predict(input_text, model, tokenizer)
-        print("output" , output)
+        print("output starcoder python" , output)
     if(model_choice==0):
         if(javaFlag == "false"):
-            print("Inside starcoder for python")
+            print("Inside starcoder for java")
             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)
+            print("output starcoder java" , output)
     else:
       a_variable = model_box[model_choice]
       output = a_variable(input_text)
+      print("output other" , output)
     return(output)
 
 
 def run_predict(text, model, tokenizer):
     prompt = text
-    pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=300)
+    pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=400)
     result = pipe(f"<s>[INST] {prompt} [/INST]")
     arr = result[0]['generated_text'].split('[/INST]')
     return arr[1]