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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -46,12 +46,12 @@ def the_process(input_text, model_choice):
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")
52
- model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
53
- output = run_predict(input_text, model, tokenizer)
54
- print("output starcoder java" , output)
55
  else:
56
  a_variable = model_box[model_choice]
57
  output = a_variable(input_text)
 
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")
52
+ model = AutoModelForCausalLM.from_pretrained("nadiamaqbool81/starcoderbase-1b-hf")
53
+ output = run_predict(input_text, model, tokenizer)
54
+ print("output starcoder java" , output)
55
  else:
56
  a_variable = model_box[model_choice]
57
  output = a_variable(input_text)