nadiamaqbool81 commited on
Commit
fd16522
·
1 Parent(s): 94dac17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -48,15 +48,15 @@ def the_process(input_text, model_choice):
48
  pythonFlag = "true"
49
  elif(pythonFlag == "true"):
50
  print("pythonFlag", pythonFlag)
51
- if(model_choice==0):
52
- if(javaFlag == "false"):
53
  print("Inside llama for python")
54
  tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
55
  model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
56
  output = run_predict(input_text, model, tokenizer)
57
  print("output" , output)
58
  javaFlag = "true"
59
- elif(javaFlag == "true"):
60
  print("javaFlag", javaFlag)
61
  else:
62
  a_variable = model_box[model_choice]
 
48
  pythonFlag = "true"
49
  elif(pythonFlag == "true"):
50
  print("pythonFlag", pythonFlag)
51
+ if(model_choice==0):
52
+ if(javaFlag == "false"):
53
  print("Inside llama for python")
54
  tokenizer = AutoTokenizer.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
55
  model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
56
  output = run_predict(input_text, model, tokenizer)
57
  print("output" , output)
58
  javaFlag = "true"
59
+ elif(javaFlag == "true"):
60
  print("javaFlag", javaFlag)
61
  else:
62
  a_variable = model_box[model_choice]