Spaces:
Runtime error
Runtime error
Commit
·
fe7f36f
1
Parent(s):
9e6998b
Update app.py
Browse files
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 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
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)
|