nadiamaqbool81 commited on
Commit
b47d696
·
1 Parent(s): fe7f36f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def the_process(input_text, model_choice):
45
  model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf_python")
46
  output = run_predict(input_text, model, tokenizer)
47
  print("output starcoder python" , output)
48
- if(model_choice==0):
49
  if(javaFlag == "false"):
50
  print("Inside starcoder for java")
51
  tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
 
45
  model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf_python")
46
  output = run_predict(input_text, model, tokenizer)
47
  print("output starcoder python" , output)
48
+ elif(model_choice==0):
49
  if(javaFlag == "false"):
50
  print("Inside starcoder for java")
51
  tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")